题目描述(简单难度)

给一个数字,将这个数字的各个位取平方然后相加,得到新的数字再重复这个过程。如果得到了 1 就返回 true,如果得不到 1 就返回 false

解法一

之前秋招的一道笔试题,当时想法也很简单。如果过程中得到了 1 直接返回 true

什么时候得不到 1 呢?产生了循环,也就是出现的数字在之前出现过,那么 1 一定不会得到了,此时返回 false

在代码中,我们只需要用 HashSet 去记录已经得到的数字即可。

public boolean isHappy(int n) { HashSet<Integer> set = new HashSet<>(); set.add(n); while (true) { int next = getNext(n); if (next == 1) { return true; } if (set.contains(next)) { return false; } else { set.add(next); n = next; } } } //计算各个位的平方和 private int getNext(int n) { int next = 0; while (n > 0) { int t = n % 10; next += t * t; n /= 10; } return next; } 

还有一个问题,代码中我们用了 while 循环,那么有没有可能永远不产生 1 并且不产生重复的数字,然后使得代码变成死循环呢?

不需要担心,因为根据我们的算法,产生的数字一定是有限的。即使产生的数字不是有限的,因为我们用的是 int 来保存数字,int 所表示的数字个数是有限的。因此,如果产生的数字是 n 个,如果我们循环到第 n + 1 次,根据鸽巢原理,此时一定会产生一个重复数字了,从而跳出 while 循环。

解法二

参考 这里-space-and-no-magic-math-property-involved-)),优化了空间复杂度到 O(1)

回想一下 141 题,判断一个链表是否有环。

而这道题,其实本质上就是判断链表是否有环,当出现重复的数字也就是产生了环。

所以我们可以用快慢指针的方法,或者叫 Floyd Cycle detection algorithm。

原理也很好理解,想象一下圆形跑道,两个人跑步,如果一个人跑的快,一个人跑的慢,那么不管两个人从哪个位置出发,跑的过程中两人一定会相遇。

所以这里我们用两个指针 fastslowfast 每次走两步,slow 每次走一步。

如果有重复的数字的话,fastslow 就一定会相遇。

没有重复数字的话,当 fast 经过 1 的时候,就会停下来了。然后 slow 最终也会走向 1,所以也会相遇。

因此,代码的话,当 fastslow 相遇的时候只需要判断当前是否是 1 即可。

public boolean isHappy(int n) { int slow = n; int fast = n; do { slow = getNext(slow); fast = getNext(getNext(fast)); } while (slow != fast); return slow == 1; } private int getNext(int n) { int next = 0; while (n > 0) { int t = n % 10; next += t * t; n /= 10; } return next; } 

解法一很常规,解法二的话将模型归结到有环链表太厉害了,自愧不如。

results matching ""

    No results matching ""