快乐数
编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」 定义为:
- 对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。 - 如果这个过程 结果为 1,那么这个数就是快乐数。
如果 n 是 快乐数 就返回 true ;不是,则返回 false 。
示例 1:
输入:n = 19
输出:true
解释:
1^2 + 9 ^2 = 82
8^2 + 2 ^2 = 68
6^2 + 8 ^2 = 100
1^2 + 0 ^2 + 0 ^2 = 1
解题思路
- 检测循环:使用集合(HashSet)来记录已经出现过的数字。
- 如果某个数字再次出现,说明进入了无限循环,不是快乐数。
- 如果存在循环且循环的数不为1,则不是快乐数。
Java实现
public class HappyNumber {
public boolean isHappy(int n) {
Set<Integer> seen = new HashSet<>();
while (n != 1 && !seen.contains(n)) {
seen.add(n);
n = getNext(n);
}
return n == 1;
}
private int getNext(int n) {
int sum = 0;
while (n > 0) {
int digit = n % 10;
n = n / 10;
sum += digit * digit;
}
return sum;
}
public static void main(String[] args) {
HappyNumber happyNumber = new HappyNumber();
// Test Case 1
int n1 = 19;
// Expected: true
System.out.println("Result: " + happyNumber.isHappy(n1));
// Test Case 2
int n2 = 123;
// Expected: false
System.out.println("Result: " + happyNumber.isHappy(n2));
}
}
时间空间复杂度
-
时间复杂度: 在最坏的情况下,每个数字在集合中最多出现一次,时间复杂度为 O(logn),其中 n 是输入数字。由于每次计算平方和时数字变小,所以不会有非常大的循环。
-
空间复杂度:使用了一个HashSet来存储出现过的数字,空间复杂度为 O(logn),因为数字的平方和在不断变小。