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

学习目标

• 练习并熟悉哈希表题目

---

学习内容

454. 四数相加 II(Middle)

题目链接: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

示例 1：
输入：nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出：2
解释：
两个元组如下：

1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0

示例 2：
输入：nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出：1

提示：
n == nums1.length
n == nums2.length
n == nums3.length
n == nums4.length
1 <= n <= 200
-228 <= nums1[i], nums2[i], nums3[i], nums4[i] <= 228

思路：两个有两层的嵌套循环，第一个统计 nums1 和 nums2 各个部分的和的个数，并将数据放入哈希表；第二个统计 0 - nums3[i] - nums4[j] 在哈希表中的个数。
时间复杂度：O(n^2) 空间复杂度：O(n)

解决方案：

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            for (int j = 0; j < nums2.length; j ++) {
                map.put(nums1[i] + nums2[j], map.getOrDefault(nums1[i] + nums2[j], 0) + 1);
            }
        }
        int cnt = 0;
        for (int i = 0; i < nums3.length; i++) {
            for (int j = 0; j < nums4.length; j ++) {
                if (map.keySet().contains(0 - nums3[i] - nums4[j])) {
                    cnt += map.get(0 - nums3[i] - nums4[j]);
                }
            }
        }

        return cnt;
    }
}

小结：其实是四层循环的分解，而分解成两套循环中间就需要快速找到第一层的结果，那么就需要引入哈希表。

383. 赎金信(Easy)

题目链接:383. 赎金信

题目：给你两个字符串：ransomNote 和 magazine ，判断 ransomNote 能不能由 magazine 里面的字符构成。
如果可以，返回 true ；否则返回 false 。
magazine 中的每个字符只能在 ransomNote 中使用一次。

示例 1：
输入：ransomNote = “a”, magazine = “b”
输出：false
示例 2：
输入：ransomNote = “aa”, magazine = “ab”
输出：false
示例 3：
输入：ransomNote = “aa”, magazine = “aab”
输出：true

提示：
1 <= ransomNote.length, magazine.length <= 105
ransomNote 和 magazine 由小写英文字母组成

思路：遍历magazine并记录下来其各个字母数，再遍历ransomNote，每遍历一个ransomNote的元素，就减少对应记录下的元素数。若没有可减的，就返回 false;
时间复杂度：O(N+M) 空间复杂度：O(1)

解决方案：

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

小结：简单题，用于复习昨天的内容。

15. 三数之和(Middle)

题目链接:15. 三数之和

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

示例 1：
输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。


示例 2：
输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。


示例 3：
输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

思路：最先想到的是双指针法。
时间复杂度：O(n^2) 空间复杂度：懒分析，大致是O(n)

解决方案：

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

                    while (left < right && nums[left] == nums[left + 1]) left ++;
                    while (left < right && nums[right] == nums[right - 1]) right --;

                    left ++;
                    right --;
                }
            }
        }

        return ans;
    }
}

小结：这道题大多是用排序+双指针的做法，不太有认同使用哈希表的解法。

18. 四数之和(Middle)

题目链接: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
你可以按 任意顺序 返回答案 。

示例 1：
输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2：
输入：nums = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

提示：
1 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109

思路：类似15. 三数之和, 多嵌套一层。
时间复杂度：O(n^3) 空间复杂度：大致是O(n)

解决方案：

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

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

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

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

小结：思路是没什么问题的，但是如Carl哥所说：难在去重和剪枝，出错点如下：

1. if (nums[i] > 0 && nums[i] > target) break; 此处因惯性思维，错写为if (nums[i] > target) break;
2. if (j > i + 1 && nums[j] == nums[j-1]) continue;此处因惯性思维，错写为if (j > 1 && nums[j] == nums[j-1]) continue;
3. 在 for(int j = i + 1; j <= nums.length - 1; j ++){...}中错误地添加了if (i > 0 && nums[j] == nums[j-1]) continue;, 导致缺少一部分合理的解。

---

总结

今日的题目前两道是传统的哈希表题目，后两道则更有些双指针的味道。而上面的15. 三数之和、18. 四数之和对于我而言，原本以为是简单多一个嵌套，没想到重点是去重和剪枝。这也警告我，相似的问题，哪怕思路一致，但是边界的确定往往更重要。