Leetcode 18.四数之和 C++

Leetcode 18.四数之和

题目

给定一个包含 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]
]

题解

此题与三数之和的核心思想相同，利用二分查找确定两个数。因此，此题中我们需要二次循环确定两个数，然后再利用二分法查找剩下的两个数。
我们首先对数组进行排序。外层循环确定最大的数，内层循环确定最小的数，然后利用二分查找的思想确定中间的两个数。即我们假设a为最小数，b为次小数，令a=nums[i],d=nums[j]，其中i<j，之后令x=j+1,y=n-1,nums[x]、nums[y]分别作为c、d。如果此时的和大于target，则y–；此时的和小于target，则x++。为了避免重复的四元组出现，我们要把a、b、c、d可能重复取的值略掉，详细方法见代码

代码

vector<vector<int>> fourSum(vector<int>& nums, int target) {
            vector<vector<int>> ans;
            sort(nums.begin(),nums.end());
            int n=nums.size(),i,j,y,x;
            for(i=0; i<n-3; ){
                for(j=i+1; j<n-2; ){
                    x=j+1;  y=n-1;
                    while(x < y){
                        int now = nums[i]+nums[j]+nums[x]+nums[y];
                        if(now == target){
                            if(x == j+1 || nums[x]!=nums[x-1]){
                                vector<int> num = {nums[i],nums[j],nums[x],nums[y]};
                                ans.push_back(num);
                            }
                            do{x++;}while(nums[x]==nums[x-1] && x<y);		//避免c取重复的值
                            do{y--;}while(nums[y]==nums[y+1] && x<y);		//避免d取重复的值
                        }else if(now > target){
                            do{y--;}while(nums[y]==nums[y+1] && x<y);		//可以用y--;一句代替
                        }else{
                            do{x++;}while(nums[x]==nums[x-1] && x<y);		//可以用x++;一句代替
                        }
                    }
                    do{j++;}while(nums[j-1] == nums[j] && j<n-2);			//避免b取重复的值
                }
                do{i++;}while(nums[i-1] == nums[i] && i<n-3);			//避免a取重复的值
            }
            return ans;
        }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。