给定一个包含 n 个整数的数组 nums
和一个目标值 target
,判断 nums
中是否存在四个元素 a,b,c 和 d ,使得 a + b + c +d 的值与 target
相等?找出所有满足条件且不重复的四元组。
注意:
答案中不可以包含重复的四元组。
示例:
给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。 满足要求的四元组集合为: [ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2] ]
此题与15题求三数之和和16题求与三数之和最接近的思路是一样,都要转化为求两数之和的问题,降低复杂度。
class Solution { public: vector<vector<int>> fourSum(vector<int>& nums, int target) { vector<vector<int>> vec; vector<int> v(4); if(nums.size()<4) return vec; sort(nums.begin(),nums.end()); for(int i=0;i<nums.size()-3;++i){ for(int j=i+1;j<nums.size()-2;++j){ int m=target-nums[i]-nums[j];//转化成求两数之和的问题,接下来的解法与求三数之和时一样 int l=j+1,r=nums.size()-1; while(l<r){ int sum=nums[l]+nums[r]; if(m==sum){ v[0]=nums[i]; v[1]=nums[j]; v[2]=nums[l]; v[3]=nums[r]; vec.push_back(v); ++l; --r; while(l<r&&nums[r]==nums[r+1]) --r;//注意去重的位置在此 while(l<r&&nums[l]==nums[l-1]) ++l; } else if(m<sum){ --r; } else{ ++l; } } while(j<nums.size()-2&&nums[j]==nums[j+1]) ++j; } while(i<nums.size()-3&&nums[i]==nums[i+1]) ++i; } return vec; } };