代码随想录刷题Day7 | 454.四数相加II | 383. 赎金信 | 15. 三数之和 | 18. 四数之和

代码随想录刷题Day7 | 454.四数相加II | 383. 赎金信 | 15. 三数之和 | 18. 四数之和

454.四数相加II

题目： 给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是n，请你计算有多少个元组 (i, j, k, l) 能满足：

• 0 <= i, j, k, l < n
• nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

思路： 本题暴力解法，遍历四个数组的话时间复杂度为O(n^{4)，加入我们将其中一个数组放到哈希表中，时间复杂度仍然还有O(n}3)，我们对这四个数组两两求和，将前两个数组求和的结果存入到哈希表中，key为值，value为出现的次数。然后计算剩下的两个数组的和并取负，然后count+哈希表中对应的value即可。

代码：

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int count = 0;
        Map<Integer, Integer> map = new HashMap<>();
        for(int num1 : nums1){
            for(int num2 : nums2){
                int tmp = num1 + num2;
                if(map.containsKey(tmp)) map.put(tmp, map.get(tmp) + 1);
                else map.put(tmp, 1);
            }
        }
        for(int num3 : nums3){
            for(int num4 : nums4){
                int tmp = 0 - (num3 + num4);
                if(map.containsKey(tmp)) count += map.get(tmp);
            }
        }
        return count;
    }
}

时间复杂度：O(n^2)

383. 赎金信

题目： 给你两个字符串：ransomNote 和 magazine ，判断 ransomNote 能不能由 magazine 里面的字符构成。

如果可以，返回 true ；否则返回 false 。

magazine 中的每个字符只能在 ransomNote 中使用一次。

思路： 因为只包含小写字母，我们考虑使用数组来进行存储。题目是判断 ransomNote 能不能由 magazine 里面的字符构成，因此先遍历magazine，对应的字母位置+1，然后遍历ransomNote，字母对应位置-1，如果出现了小于0以下的数值，那么返回false，待遍历完两个字符串没有小于0的情况出现，返回true。

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

代码：

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

时间复杂度O(n)，空间复杂度O(1)

15. 三数之和

题目： 给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足i != j、i != k 且 j != k，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。

注意： 答案中不可以包含重复的三元组。

思路： 该题不适用哈希表，剪枝操作十分麻烦，应用双指针来做，首先对数组排序，然后用一个for循环，让i从下标为0的地方开始，同时定义一个left为i+1，right在数组最末尾的位置，然后判断nums[i] + nums[left] + nums[right]的值大于0，right--，该值小于0，left++，等于0，则符合我们的要求。下面是剪枝的步骤：

1. 对于i，如果num[i]==nums[i-1]，则说明第一个值重复，直接进入下一次循环
2. 因为我们事先已经对于数组排序，因此如果nums[i]>0直接退出循环，返回结果
3. 对于left和right，也要避免重复元素，因此当和为0时，重复的left和right直接跳过

代码：

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> result = new ArrayList<>();
        for(int i = 0; i < nums.length - 2; i++){
            if(nums[i] > 0) break;
            if(i > 0 && nums[i] == nums[i - 1]) continue;
            int left = i + 1, right = nums.length - 1;
            while(right > left){
                int sum = nums[i] + nums[left] + nums[right];
                if(sum > 0) right--;
                else if(sum < 0) left++;
                else{
                    result.add(Arrays.asList(nums[i], nums[left], nums[right]));
                    while (right > left && nums[right] == nums[right - 1]) right--;
                    while (right > left && nums[left] == nums[left + 1]) left++;
                    right--;
                    left++;
                }
            }
        }
        return result;
    }
}

时间复杂度：O(n^2)

18. 四数之和

题目： 给你一个由 n 个整数组成的数组 nums ，和一个目标值target。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

• 0 <= a, b, c, d < n
• ``a、b、c和d` 互不相同
• nums[a] + nums[b] + nums[c] + nums[d] == target
  你可以按 任意顺序 返回答案 。

思路： 思路和三数之和相似，都是使用双指针，三数之和时使用一层for循环nums[i]作为确定之，四数之和时使用两层for循环nums[i]+nums[j]作为确定值。

注意下面这个用例，要将sum转化为long格式才能通过。

[1000000000,1000000000,1000000000,1000000000]
-294967296

代码：

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        List<List<Integer>> result = new ArrayList<>();
        for(int i = 0; i < nums.length - 3; i++){
            if(i > 0 && nums[i] == nums[i - 1]) continue;
            for(int j = i + 1; j < nums.length - 2; j++){
                if(j > i + 1 && nums[j] == nums[j - 1]) continue;
                int left = j + 1, right = nums.length - 1;
                while(right > left){
                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]));
                    while(right > left && nums[right] == nums[right - 1]) right--;
                    while(right > left && nums[left] == nums[left + 1]) left++;
                    right--;
                    left++;
                }
                }
            }
        }
        return result;
    }
}

时间复杂度O(n^3)

三数之和，四数之和的原理是相同的，如果有五数之和，六数之和也是相同的做法，n数之和的时间复杂度为O(n^n-1)

总结

1. 对于总类较少的数据，比如说字母，可以考虑使用数组来记录频率等
2. 遇到要经常访问的数组，不放考虑一下使用哈希表的方式是否可以操作，遍历数组的时间复杂度为O(n)，但是从哈希表中查找元素时间复杂度为O(1)
3. 三数之和，四数之和使用哈希表的方法剪枝操作十分复杂，使用双指针更简单一些。
4. 常用的哈希表为set和map，一个单列，一个双列