LeetCode刷题之哈希表 第202题 快乐数

题目描述

编写一个算法来判断一个数 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;
    }
}