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

解题思路

本题主要思路 哈希+分组！

1.先将A数组和B数组的元素通过双重循环两两相加的值作为key，然后用value统计每个key出现的次数。

2.将C数组和B数组的元素也通过双重循环两两相加后取反，到map中找是否存在该key值，存在则结果+=value。

3.最后返回结果值就可以了

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        Map<Integer,Integer>map=new HashMap<>();
        int n=nums1.length;
        int result=0;
        for(int i=0;i<n;i++){遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
            for(int j=0;j<n;j++){
                int sum=nums1[i]+nums2[j];
                if(map.containsKey(sum)){
                    map.put(sum,(map.get(sum)+1));
                }
                else{
                    map.put(sum,1);
                }
            }
        }
        for(int i=0;i<n;i++){再遍历大C和大D数组，找到0-(c+d) 在map中
            for(int j=0;j<n;j++){
                int sum=-(nums3[i]+nums4[j]);
                if(map.containsKey(sum)){
                    result+=map.get(sum);
                }
            }
        }
        return result;
    }
}

383. 赎金信

题目

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

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

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

示例 1：

输入：ransomNote = "a", magazine = "b"
输出：false

解题思路---map

这题我的思路是用map，但是用数组是更节省时间的写法。

我说说我的思路吧

先遍历magazin中的字母并统计出现次数，然后遍历ransomNote对字母进行--操作，如果为0了，或者map中没有该字母，返回false，否则返回true。

class Solution {
    public boolean canConstruct(String ransomNote, String magazine) {
        Map<Character,Integer>map=new HashMap<>();
        for(int i=0;i<magazine.length();i++){//将magazine中的字母以及出现个数统计
            char c=magazine.charAt(i);
            if(map.containsKey(c)){
                map.put(c,(map.get(c)+1));
            }
            else map.put(c,1);
        }
        for(int i=0;i<ransomNote.length();i++){//遍历ransomNote，在map里对应的字符个数做--操作
            char c=ransomNote.charAt(i);
            if(map.containsKey(c)){
                if(map.get(c)==0)return false;
                else map.put(c,(map.get(c)-1));
            }
            else 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 且不重复的三元组。

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

示例 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] 。
注意，输出的顺序和三元组的顺序并不重要。

解题思路---双指针法

拿这个nums数组来举例，首先将数组排序，然后有一层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) {
        List<List<Integer>> ans=new ArrayList<List<Integer>>();
        Arrays.sort(nums);

        for(int i=0;i<nums.length;i++){

            if(nums[i]>0)return ans;
            if(i>0&&nums[i]==nums[i-1])continue;

            int left=i+1;
            int 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 ans.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--;
                
                right--; 
                left++;
            }
        }
        return ans;
    }
}

18. 四数之和

题目

给你一个整数数组 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] 。
注意，输出的顺序和三元组的顺序并不重要。

解题思路

和三数之和几乎一样，有两点不同

1.剪枝条件不同，不要判断nums[k] > target 就返回了，三数之和 可以通过 nums[i] > 0 就返回了，因为 0 已经是确定的数了，四数之和这道题目 target是任意值。比如：数组是[-4, -3, -2, -1]，target是-10，不能因为-4 > -10而跳过。但是我们依旧可以去做剪枝，逻辑变成nums[i] > target && (nums[i] >=0 || target >= 0)就可以了。

2.固定了i和k两个点

代码如下：

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> ans=new ArrayList<List<Integer>>();
        Arrays.sort(nums);
        for(int i=0;i<nums.length;i++){
            if(nums[i]>target&&nums[i]>=0)break;
            if(i>0&&nums[i]==nums[i-1])continue;

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

                while(right>left){
                    long sum=(long)nums[i]+nums[k]+nums[left]+nums[right];
                    if(sum>target)right--;
                    else if(sum<target)left++;
                    else {
                        ans.add(Arrays.asList(nums[i],nums[k],nums[left],nums[right]));

                        while(right>left&&nums[left]==nums[left+1])left++;
                        while(right>left&&nums[right]==nums[right-1])right--;
                        
                        right--; 
                        left++;
                    }
                }
            }
        }
        return ans;

    }
}