代码随想录算法训练营第六天| 454.四数相加II、383. 赎金信、15. 三数之和、18. 四数之和、总结

454.四数相加II 

建议：本题是 使用map 巧妙解决的问题，好好体会一下 哈希法 如何提高程序执行效率，降低时间复杂度，当然使用哈希法 会提高空间复杂度，但一般来说我们都是舍空间 换时间， 工业开发也是这样。

题目链接/文章讲解/视频讲解：代码随想录

没思路

四数相加与四数之和（一个数组中找出四个元素相加）相比，不用去重（不同位置上的相同元素视为不同的）

先遍历两个，后遍历两个，O(n^2+n^2)=O(n^2)；先遍历一个，后遍历三个，O(n+n^3)=O(n^3)

本题解题步骤：

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

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        Map<Integer, Integer> map = new HashMap<>();

        int count = 0;

        for(int i = 0; i < nums1.length; i++){
            for(int j = 0; j< nums2.length; j++){
                int temp = nums1[i]+nums2[j];
                if(map.containsKey(temp)){
                    map.put(temp, map.get(temp)+1);
                }else{
                    map.put(temp, 1);
                }
            }
        }

        for(int i = 0; i < nums3.length; i++){
            for(int j = 0; j< nums4.length; j++){
                int temp = -(nums3[i]+nums4[j]);

                if(map.containsKey(temp)){
                    count += map.get(temp);
                }
            }
        }
        return count;
    }
}

---

383. 赎金信  

建议：本题 和 242.有效的字母异位词 是一个思路 ，算是拓展题 

题目链接/文章讲解：代码随想录

因为题目所只有小写字母，那可以采用空间换取时间的哈希策略， 用一个长度为26的数组还记录magazine里字母出现的次数。

然后再用ransomNote去验证这个数组是否包含了ransomNote所需要的所有字母。

依然是数组在哈希法中的应用。

在本题的情况下，使用map的空间消耗要比数组大一些的，因为map要维护红黑树或者哈希表，而且还要做哈希函数，是费时的！数据量大的话就能体现出来差别了。 所以数组更加简单直接有效！

class Solution {
    public boolean canConstruct(String ransomNote, String magazine) {
        // shortcut
        if (ransomNote.length() > magazine.length()) {
            return false;
        }

        int[] record = new int[26];

        for(char i: magazine.toCharArray()){
            record[i-'a']++;
        }

        for(char i: ransomNote.toCharArray()){
            record[i-'a']--;
        }

        for(int i: record){
            if(i < 0){
                return false;
            }
        }
        return true;
    }
}

---

15. 三数之和 

建议：本题虽然和 两数之和 很像，也能用哈希法，但用哈希法会很麻烦，双指针法才是正解，可以先看视频理解一下 双指针法的思路，文章中讲解的，没问题 哈希法很麻烦。 

题目链接/文章讲解/视频讲解：代码随想录

由于要去重（a、b、c互不相等），所以哈希法较复杂（在寻找c时没什么顺序，去重较麻烦），因此此题不建议用哈希法。

使用双指针法一定要先对数组进行排序。本题要求找到元素值，没有要求返回下标（要求返回下标，则下标会乱，不能排序），所以可以排序。

为什么要排序？排序后带来了哪些便利？

由于是排完序的，

• 如果三数和>0，只有让right向左移一位才能让三数和缩小，所以right--；
• 如果三数和<0，只有让left向右移一位才能让三数和增大，所以left++
• 如果三数和=0，收获结果，将nums[i]、nums[left]、nums[right]加入result数组（二维数组，存放可能的符合条件的结果集）

注意，nums[i]，nums[left]，nums[right]要做去重，因为结果集里不能有重复的三元素。

a 如果重复了怎么办，a是nums里遍历的元素，那么应该直接跳过去。

但这里有一个问题，是判断 nums[i] 与 nums[i + 1]是否相同，还是判断 nums[i] 与 nums[i-1] 是否相同。

if (nums[i] == nums[i + 1]) { // 去重操作，实际上是结果集里不能有重复元素
    continue;
}

这样会把三元组中出现重复元素的情况直接pass掉了。 例如{-1, -1 ,2} 这组数据，当遍历到第一个-1 的时候，判断下一个也是-1，那这组数据就pass了。

我们要做的是不能有重复的三元组，但三元组内的元素是可以重复的！

所以这里是有两个重复的维度。因此，应该是：

if (i > 0 && nums[i] == nums[i - 1]) {
    continue;
}

这么写就是当前使用 nums[i]，我们判断前一位是不是一样的元素，在看 {-1, -1 ,2} 这组数据，当遍历到 第一个 -1 的时候，只要前一位没有-1，那么 {-1, -1 ,2} 这组数据一样可以收录到结果集里。 

对b和c去重，需要在将a、b、c加入结果集后（否则存在还没收获结果，已经left>right的情况），

• 如果nums[right] == nums[right - 1]持续移动right，
• 如果nums[left] == nums[left + 1]持续移动left，

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);

        for(int i = 0; i < nums.length; i++){
            if(nums[i] > 0){
                return result;
            }
            if(i > 0 && nums[i] == nums[i-1]) continue;
            int left = i + 1;
            int right = nums.length - 1;
            while(right > left){
                if(nums[i] + nums[left]+ nums[right] > 0){
                    right--;
                }else if(nums[i] + nums[left]+ nums[right] < 0){
                    left++;
                }else{
                    result.add(Arrays.asList(nums[i], nums[left], nums[right]));

                    // 去重逻辑应该放在找到一个三元组之后，对b 和 c去重
                    while(right > left && nums[right] == nums[right - 1]) right--;
                    while(right > left && nums[left] == nums[left + 1]) left++;

                    right--; 
                    left++;
                }
            }
        }

        return result;
        
    }
}

---

18. 四数之和  

建议： 要比较一下，本题和 454.四数相加II 的区别，为什么 454.四数相加II 会简单很多，这个想明白了，对本题理解就深刻了。 本题 思路整体和 三数之和一样的，都是双指针，但写的时候 有很多小细节，需要注意，建议先看视频。 

题目链接/文章讲解/视频讲解：代码随想录

相较于三数之和，

• 多一层for循环，一次固定两个元素，还是用left和right两个指针
• 相加等于target是输入的值，不再是固定的0
• 不要判断nums[k] > target 就返回了，三数之和 可以通过 nums[i] > 0 就返回了，因为 0 已经是确定的数了，四数之和这道题目 target是任意值。比如：数组是[-4, -3, -2, -1]，target是-10，不能因为-4 > -10而跳过。但是我们依旧可以去做剪枝，逻辑变成nums[i] > target && (nums[i] >=0 || target >= 0)就可以了。

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);

        for(int i = 0; i < nums.length; i++){
            // if(nums[i] > target) return result; 不能这么剪枝，因为target有可能是负数
            if(nums[i] > target && target >= 0) return result;
            if(i > 0 && nums[i] == nums[i-1]) continue;

            for(int j = i+1; j < nums.length; j++){
                if(j > i+1 && nums[j] == nums[j-1]) continue;

                //接下来和三数之和逻辑一样
                int left = j + 1;
                int right = nums.length - 1;
                while(right > left){
                    // nums[k] + nums[i] + nums[left] + nums[right] > target int会溢出
                    long sum = (long) nums[i] + nums[j] + nums[left] + nums[right];
                    if(sum > target){
                        right--;
                    }else if(sum < target){
                        left++;
                    }else{
                        result.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));

                        // 去重逻辑应该放在找到一个三元组之后，对b 和 c去重
                        while(right > left && nums[right] == nums[right - 1]) right--;
                        while(right > left && nums[left] == nums[left + 1]) left++;

                        right--; 
                        left++;
                    }
                }
            }
        }
        return result;
    }
}

四数之和的双指针解法是两层for循环nums[k] + nums[i]为确定值，依然是循环内有left和right下标作为双指针，找出nums[k] + nums[i] + nums[left] + nums[right] == target的情况，三数之和的时间复杂度是O(n^2)，四数之和的时间复杂度是O(n^3) 。

那么一样的道理，五数之和、六数之和等等都采用这种解法。

对于15.三数之和 (opens new window)双指针法就是将原本暴力O(n^3)的解法，降为O(n^2)的解法，四数之和的双指针解法就是将原本暴力O(n^4)的解法，降为O(n^3)的解法。

之前我们讲过哈希表的经典题目：454.四数相加II (opens new window)，相对于本题简单很多，因为本题是要求在一个集合中找出四个数相加等于target，同时四元组不能重复。

而454.四数相加II (opens new window)是四个独立的数组，只要找到A[i] + B[j] + C[k] + D[l] = 0就可以，不用考虑有重复的四个元素相加等于0的情况，所以相对于本题还是简单了不少