代码随想录day5

哈希表

快乐数

编写一个算法来判断一个数 n 是不是快乐数。

「快乐数」 定义为：

• 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
• 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
• 如果这个过程 结果为 1，那么这个数就是快乐数。

如果 n 是 快乐数 就返回 true ；不是，则返回 false 。

思路

1、无限循环其实就是找环的思路，可以借用链表中判断有无环的快慢指针方法
2、借助于集合判断是否有重复元素，若重复，则说明已进入循环

class Solution {
    public boolean isHappy(int n) {
        if(n == 1) return true;
        Set<Integer> set = new HashSet<>();
        set.add(n);
        while(true){
            int next = 0;
            // 求解next
            while(n > 0){
                next += (n % 10) * (n % 10);
                n = n / 10;
            }
            if(next == 1)
                return true;
            // 是否重复
            if(set.contains(next))
                break;
            set.add(next);
            n = next;
        }
        return false;

    }
}

//第二种快慢指针
class Solution {
    // 利用“快慢指针思想，找出循环”
    public boolean isHappy(int n) {
        int slow = n,fast = numSquaue(n);
        while(slow != fast){
            slow = numSquaue(slow);
            fast = numSquaue(numSquaue(fast));
        }

        return slow == 1;
        
    }   
    int numSquaue(int n){
        int sum = 0;
        while(n > 0){
            int bit = n % 10;
            sum += bit * bit;
            n = n / 10;
        }
        return sum;
    }
}

两数之和

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。
示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

思路

显而易见的是采用暴力解法可以求解，但时间复杂度过高，如何优化？
借助于Map集合，key - 值，value - 下标，在一次遍历中即可求解

class Solution {
    // 使用哈希表优化
    public int[] twoSum(int[] nums, int target) {
        int n = nums.length;
        Map<Integer,Integer> map = new HashMap<>();
        for(int i = 0;i < n;i++){
            if(map.containsKey(target - nums[i]))
                return new int[]{map.get(target - nums[i]),i};
            map.put(nums[i],i);
        }
        return new int[0];
    }
}