leetcode 18.四数之和

1.题目

给你一个由 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.示例

1)示例 1:
输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

2)示例 2:
输入:nums = [2,2,2,2,2], target = 8
输出:[[2,2,2,2]]

3.数据范围

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

4.分析

这道题目是三数之和的进阶版,多了一个数,将原本的单层循环改为外围双层循环枚举前两个数a,b即可,且四个数满足aa+1且和前一个点相同,这样说明也已经枚举了b作为第二小的情况了,c,d排除重复同理。在此基础上,需要注意tarhet的范围在计算a+b+c+d时候可能会出现超出int的范畴,且虽然定义了longlong sum=a+b+c+d,但计算时仍然是先存在int类型中的,会报错:runtime error: signed integer overflow,可以先把a,b,c,d都转化为long long 形式的就可以解决问题了。

5. 代码

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ans;
        if(nums.size()<4) return ans;
        sort(nums.begin(),nums.end());
        // ans.push_back(nums);
        for(int i=0;i<nums.size()-3;i++){
            if(i>0){
                if(nums[i]==nums[i-1]) continue;
            }
            for(int j=i+1;j<nums.size()-2;j++){
                if(j>i+1&&nums[j]==nums[j-1]) continue;
                int left=j+1,right=nums.size()-1;
                while(left<right){
                    long long a=nums[i],b=nums[j],c=nums[left],d=nums[right];
                    long long sum=a+b+c+d;
                    if(sum==target){
                        vector<int> test;
                        test.push_back(nums[i]);
                        test.push_back(nums[j]);
                        test.push_back(nums[left]);
                        test.push_back(nums[right]);
                        ans.push_back(test);
                        while(left<right&&nums[left]==nums[left+1]) left++;
                        while(left<right&&nums[right]==nums[right-1]) right--;
                    } 
                    if(sum>target) right--;
                    else left++;
                }
            }
        }
        return ans;
    }
};

你可能感兴趣的:(leetcode,leetcode,算法,职场和发展)