代码随想录算法训练营第六天| 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

思路：

思路与《1、两数之和》类似。nums1与nums2的两数之和作为第一个数a，nums3与nums4的两数之和作为第二个数b，需要找出a+b=0的数量。

定义一个map集合，key用于存放数组nums1和nums2的两数之和，value存放key出现的次数。

遍历数组nums3和nums4，查找map集合中等于0 - (nums3[i] + nums4[j])的key，并统计对应的value。

代码：

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int count = 0;
        int n = nums1.length;
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                int key = nums1[i] + nums2[j];
                if (!map.containsKey(key)) {
                    map.put(key, 1);
                } else {
                    int num = map.get(key);
                    map.put(key, num + 1);
                }
            }
        }
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                int key = 0 - (nums3[i] + nums4[j]);
                if (map.containsKey(key)) {
                    count = count + map.get(key);
                }
            }
        }
        return count;
    }
}

383. 赎金信

题目：

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

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

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

1 <= ransomNote.length, magazine.length <= 105

ransomNote 和 magazine 由小写英文字母组成

思路：

思路与《242.有效的字母异位词》类似，利用数组记录26种字符的数量。

代码：

class Solution {
    public boolean canConstruct(String ransomNote, String magazine) {
        int[] arr = new int[26];
        for (int i = 0; i < magazine.length(); i++) {
            arr[magazine.charAt(i) - 'a']++;
        }
        for (int i = 0; i < ransomNote.length(); i++) {
            arr[ransomNote.charAt(i) - 'a']--;
        }
        for (int num : arr) {
            if (num < 0) {
                return false;
            }
        }
        return true;
    }
}

15. 三数之和

题目：

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

思路：

双指针法：三个数字，每次固定一个值，两个指针指向另外两个值，遍历所有三数之和的情况。

首先给数组中的所有元素从小到大排序。遍历数组，固定值为nums[i]，每次left指针初始指向索引i+1的位置，right指针初始指向索引nums.length-1的位置。

计算三数之和sum = nums[i] + nums[left] + nums[right]。

若sum>0，则需要减小sum，right--；

若sum<0，则需要增大sum，left++；

若sum=0，则将三元组存入result中，此时单独移动一个指针在不重复的情况下sum必不为0，因此需要同时移动两个指针（right--  left++）。

题目要求答案中不可以包含重复的三元组，因此需要考虑去重。

去重只需要对三个数字分别考虑即可：

①当固定值nums[i]与前一轮循环的固定值nums[i-1]相同时，跳过此次循环；

②当left指针当前指向的值nums[left]与left下一步要指向的值nums[left+1]相等，则left++，跳过一步；

③当right指针当前指向的值nums[right]与right下一步要指向的值nums[right-1]相等，则right--，跳过一步；

代码：

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 (left < right) {
                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[left] == nums[left + 1]) {left++;}
                    while (right > left && nums[right] == nums[right - 1]) {right--;}
                    left++;
                    right--;
                }
            }
        }
        return result;
    }
}

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

你可以按 任意顺序 返回答案 。

思路：

与三数之和类似。四数之和，设置两个固定值，两个指针。

注意：这里的目标值是target不是0，判断条件要改变。

代码：

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

    }
}