代码随想录算法训练营第七天|哈希表part02|454.383.15.18

2023/8/1 任务
454.四数相加II，383. 赎金信，15. 三数之和，18. 四数之和，总结

454. 四数相加II ⭐

题目链接/文章讲解/视频讲解

思路：

1. 首先定义 一个HashMap，key放a和b两数之和，value 放a和b两数之和出现的次数。
2. 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
3. 定义int变量res，用来统计 a+b+c+d = 0 出现的次数。
4. 在遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用res把map中key对应的value也就是出现次数统计出来。
5. 最后返回统计值 res 就可以了

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i : nums1) {
            for (int j : nums2) {
                int sum = i + j;
                map.put(sum, map.getOrDefault(sum, 0) + 1);
            }
        }
        int res = 0;
        for (int i : nums3) {
            for (int j : nums4) {
                res += map.getOrDefault(0 - i - j, 0);
            }
        }
        return res;
    }
}

时间复杂度: O(n^2)
空间复杂度: O(n^2)

383. 赎金信

思路：
类似242.有效的字母异位词。

题目链接/文章讲解/视频讲解

class Solution {
    public boolean canConstruct(String ransomNote, String magazine) {
        if (ransomNote.length() > magazine.length()) {
            return false;
        }
        int[] record = new int[26];
        for (char c : ransomNote.toCharArray()) {
            record[c - 'a'] += 1;
        }
        for (char c : magazine.toCharArray()) {
            record[c - 'a'] -= 1;
        }
        for (int i : record) {
            if (i > 0) {
                return false;
            }
        }
        return true;
    }
}

15. 三数之和 ⭐⭐

双指针法

题目链接/文章讲解/视频讲解

思路：
i为第一个数，left和right分别为剩下两个数。

首先将数组排序，然后有一层for循环，i从下标0的地方开始，同时定一个下标left 定义在i+1的位置上，定义下标right 在数组结尾的位置上。

依然还是要在数组中找到 abc 使得a + b + c =0，我们这里相当于 a = nums[i]，b = nums[left]，c = nums[right]。

接下来如何移动left 和right呢， 如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了，因为数组是排序好了的，所以right下标就应该向左移动，这样才能让三数之和小一些。

如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了，left 就向右移动，才能让三数之和大一些，直到left与right相遇为止。

注意：
三处去重方法

class Solution {
   public List<List<Integer>> threeSum(int[] nums) {
       Arrays.sort(nums);
       List<List<Integer>> res = new ArrayList<>();
       for (int i = 0; i < nums.length; i++) {
           // 如果最小的数都大于0，三数之和一定大于0
           if (nums[i] > 0) {
               return res;
           }
           // 对第一个数去重：如果当前数和上个数相同，就跳过
           if (i > 0 && nums[i] == nums[i - 1]) {
               continue;
           }
           int left = i + 1;
           int right = nums.length - 1;
           while (left < right) {
               int sum = nums[i] + nums[left] + nums[right];
               if (sum > 0) {
                   right--;
               } else if (sum < 0) {
                   left++;
               } else {
                   res.add(Arrays.asList(nums[i], nums[left], nums[right]));
                   // left和right去重
                   while (left < right && nums[right] == nums[right - 1]) {
                       right--;
                   }
                   while (left < right && nums[left] == nums[left + 1]) {
                       left++;
                   }
                   // 找到答案时，双指针同时收缩
                   right--;
                   left++;
               }
           }

       }
       return res;
   }
}

时间复杂度: O(n^2)
空间复杂度: O(1)

虽然它类似于两数之和，但两数之和不能使用双指针法，因为它要求返回的是索引下标， 而双指针法一定要排序，一旦排序之后原数组的索引就被改变了。

18. 四数之和

想明白和四数相加的区别。

思路：
在三数之和的基础上套个for循环，重点在于剪枝和去重操作。在三数之和中，剪枝就是在nums[i] > 0的时候结束，但是这道题的target可以为负数，如果只是判断nums[i] > target就结束就错了，比如target为-10，nums为{-4,-3,-2,-1}，因此需要加上nums[i] > 0。

题目链接/文章讲解/视频讲解

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
        	// 剪枝操作！！！
            if (nums[i] > 0 && nums[i] > target) {
                return res;
            }
            // 对 i 去重
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            for (int j = i + 1; j < nums.length; j++) {
            	// 对 j 去重
                if (j > i + 1 && nums[j] == nums[j - 1]) {
                    continue;
                }
                int left = j + 1;
                int right = nums.length - 1;
                while (left < right) {
                	// int会溢出！
                    long sum = nums[i] + nums[j] + nums[left] + nums[right];
                    if (sum > target) {
                        right--;
                    } else if (sum < target) {
                        left++;
                    } else {
                        res.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
                        // 对 left 去重
                        while (left < right && nums[left] == nums[left + 1]) {
                            left++;
                        }
                        // 对 right去重
                        while (left < right && nums[right] == nums[right - 1]) {
                            right--;
                        }
                        right--;
                        left++;
                    }
                }
            }

        }
        return res;
    }
}

时间复杂度: O(n^3)
空间复杂度: O(1)