18. 四数之和

Posted 2022-11-29 Ston.V

1.Description

给你一个由 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
你可以按 任意顺序 返回答案 。

2.Example

示例 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]]

3.Code(排序+双指针)

思路和之前的三数之和一样

class Solution 
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) 
        int n = nums.size();
        sort(nums.begin(),nums.end());
        vector<vector<int>> res;
        for(int i=0;i<n;i++)
            //while(i>0 && nums[i] == nums[i-1])i++;
            if(i>0 && nums[i] == nums[i-1])continue;
            for(int j=i+1;j<n;j++)
                //while(j>i+1 && nums[j] == nums[j-1])j++;
                if(j>i+1 && nums[j] == nums[j-1])continue;
                int k = j+1;
                int l = n-1;
                while(k<l)
                    //while(k<n-1 && nums[k] == nums[k-1])k++;
                    // if(k>=l)
                    //     break;
                    if((long)nums[i]+nums[j]+nums[k]+nums[l] > target)l--;
                    else if((long)nums[i]+nums[j]+nums[k]+nums[l] < target)k++;
                    else 
                        res.push_back(nums[i],nums[j],nums[k],nums[l]);
                        k++;
                        while(k<n-1 && nums[k] == nums[k-1])k++;
                    
                
            
        
        return res;
    
;

 4.注意

1.在i,j循环时，检查是否有重复元素时，我们没有while()i++来一步到位，而是直接continue，这样可以避免对临界条件的判断

2.注意检查重复元素的操作，是在上一轮循环结束后，要么没找到，要么找到后；注意对下表为K的元素的重复检查，是放在找到和等于target之后操作的；如果放在while(k<l)循环的开头，会导致在上一轮j的循环中，因为k的自增，大于了l，从而错过了答案

3.四个数相加，我们对三个下标进行重复检测即可，剩下一个自然不同。