题目描述
编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为:
对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。
如果 可以变为 1,那么这个数就是快乐数。
如果 n 是快乐数就返回 true ;不是,则返回 false 。
示例1
输入:19
输出:true
解释:
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
示例2
输入:n = 2
输出:false
方法一:哈希表检测循环
算法思想
题目中说了会 「无限循环」,那么也就是说「求和的过程中,sum会重复出现,这对解题很重要!」当我们遇到了要快速判断一个元素是否出现集合里的时候,就要考虑哈希法了。」所以这道题目使用哈希法,来判断这个sum是否重复出现,如果重复了,判断这个数是不是1,不是就return false,否则 return true
class Solution {
int getNext(int num) {
int next = 0;
while (num != 0) {
int d = num % 10;
next += d * d;
num /= 10;
}
return next;
}
public boolean isHappy(int n) {
Set<Integer> set = new HashSet<>();
if (n == 1) {
return true;
}
while (!set.contains(n)) {
set.add(n);
n = getNext(n);
}
return n == 1;
}
}
复杂度分析
时间复杂度:O(m+n),其中 m 和 n 分别是两个数组的长度。使用两个集合分别存储两个数组中的元素需要 O(m+n) 的时间,遍历较小的集合并判断元素是否在另一个集合中需要 O(min(m,n)) 的时间,因此总时间复杂度是 O(m+n)。
空间复杂度:O(m+n),其中 m 和 n 分别是两个数组的长度。空间复杂度主要取决于两个集合。
方法二:双指针
算法思想
这个问题可以转换为检测一个链表是否有环。我们不是只跟踪链表中的一个值,而是跟踪两个值,fast 和 slow。在算法的每一步中,slow 在链表中前进 1 个节点,fast 前进 2 个节点(对 getSum(n) 函数的嵌套调用)。
如果 n 是一个快乐数,即没有循环,那么 fast 最终会比 slow 先到达数字 1。
如果 n 不是一个快乐的数字,那么最终 fast 和 slow 将在同一个数字上相遇。
JAVA代码
class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
Arrays.sort(nums1);
Arrays.sort(nums2);
int i = 0, j = 0;
Set<Integer> nums = new HashSet<Integer>();
while (i < nums1.length && j < nums2.length) {
if (nums1[i] == nums2[j]) {
nums.add(nums1[i]);
} else if (nums1[i] < nums2[j]) {
++i;
} else if (nums1[i] > nums2[j]) {
++j;
}
}
int[] newArr = new int[nums.size()];
int index = 0;
for (int num : nums) {
newArr[index++] = num;
}
return newArr;
}
}