算法刷题记录.leetcode18.四数之和

74 阅读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 你可以按 任意顺序 返回答案 。

来源:力扣(LeetCode) 链接:leetcode.cn/problems/4s…

解题思路

延续三数之和解题思路,两层循环找第一二个元素,再用双指针寻找剩余两个元素。

我的代码难点

if((long)nums[i]+nums[j]+nums[left]+nums[right] > target){}

代码

class Solution {

public:

vector<vector<int>> fourSum(vector<int>& nums, int target) {

sort(nums.begin(),nums.end());

vector<vector<int>> res;

int left,right;

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

if(i>0 && nums[i]==nums[i-1]){

continue;

}

for(int j = i+1 ; j<nums.size(); j++){

if(nums[j]==nums[j-1] && j > (i+1)){

continue;

}

left = j+1;

right = nums.size()-1;

while(left<right){

if((long)nums[i]+nums[j]+nums[left]+nums[right] > target){

right--;

}else if((long)nums[i]+nums[j]+nums[left]+nums[right] < target){

left++;

}else{

res.push_back({nums[i],nums[j],nums[left],nums[right]});

while(left<right && nums[left] == nums[++left]);

while(left<right && nums[right] == nums[--right]);

}

}

}

}

return res;

}

};