day07 | 四数相加II、四数之和

目录

四数相加||

四数之和

---

四数相加||

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

思路：这道题目是四个独立的数组，只要找到A[i] + B[j] + C[k] + D[l] = 0就可以，不用考虑有重复的四个元素相加等于0的情况 

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        Map<Integer,Integer> map = new HashMap<>();
        int res = 0;
        //统计两个数组中的元素之和，同时统计出现的次数，放入map
        for(int i : nums1)
        for(int j : nums2){
        int sum = i + j;
        map.put(sum,map.getOrDefault(sum,0) + 1);
        }
        
        //统计剩余的两个元素的和，在map中找是否存在相加为0的情况，同时记录次数
        for(int i : nums3)
        for(int j : nums4)
        res += map.getOrDefault(0 - j - i , 0);
        return res;
    }
}

四数之和

题目链接：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循环以及二级剪枝操作即可

三数之和：首先要对数组进行排序操作，之后用for循环遍历得到第一个数，然后采用左右指针left、right来寻找第二、第三个数字。

 如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了，right--；

如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了，left++；

找到满足条件的三个数后，用Arrays.asList方法进行收集，直至left，right相遇，本次循环结束，进入下一次循环

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        int n = nums.length;
        Arrays.sort(nums);
        for(int k = 0; k < n - 3; k++){
            //一级剪枝操作，直接return
            if(nums[k] > target && nums[k] >= 0) return res;
            if(k > 0 && nums[k] == nums[k-1])continue;//对nums[k]去重

            for(int i = k + 1; i < n - 2; i++){
                //二级剪枝操作，要break退出内循环，外循环还得继续
                if(nums[k] + nums[i] > target && nums[k] + nums[i] >= 0) break;
                if(i > k + 1 && nums[i] == nums[i-1]) continue;//对nums[i]去重

                int left = i + 1;
                int right = n - 1;
                while(left < right){
                    int sum = nums[k] + nums[i] + nums[left] + nums[right];
                    if(sum < target)left++;
                    else if(sum > target)right--;
                    else{
                        res.add(Arrays.asList(nums[k],nums[i],nums[left],nums[right]));
                        left++;right--;
                        //对nums[left]和nums[right]去重
                        while(left < right && nums[left] == nums[left-1])left++;
                        while(left < right && nums[right] == nums[right+1])right--;
                    }
                }
            }
        }
        return res;
    }
}