四数之和-CSDN博客

本文链接：https://blog.csdn.net/qq_37499774/article/details/115767081

给定一个包含n个整数的数组nums和一个目标值target，判断nums中是否存在四个元素a,b,c,d，使得a + b + c + d = target。找出所有满足条件且不重复的四元组。

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

思路：排序 + 双指针

与三数之和相似，其实就是在前面的基础上多添加了一个遍历的指针而已。
使用四个指针（a < b < c < d）。固定最小的a和b在左边，c = b + 1，d = _size - 1，移动两个指针包夹求解。保存使得nums[a] + nums[b] + nums[c] + nums[d] == target的解。偏大时d左移，偏小时c右移。c和d相遇时，表示以当前的a和b为最小值的解已经全部求得。b++，进入下一轮循环b循环，当b循环结束后，a++，进入下一轮a循环。即a在最外层循环，里面嵌套b循环，再嵌套双指针c，d包夹求解。

准备工作：因为要使用双指针的方法，排序必须要做。时间复杂度O(NlogN).
解决重复解：移动指针时可能出现重复数字的情况。确保移动指针后，对应的数字要发生改变。

class Solution{
	public: 
	vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        vector<vector<int> > res;
        if(nums.size()<4)
        return res;
        int a,b,c,d,_size=nums.size();
        for(a=0;a<=_size-4;a++){
        	if(a>0&&nums[a]==nums[a-1]) continue;      //确保nums[a] 改变了
        	for(b=a+1;b<=_size-3;b++){
        		if(b>a+1&&nums[b]==nums[b-1])continue;   //确保nums[b] 改变了
        		c=b+1,d=_size-1;
        		while(c<d){
        			if(nums[a]+nums[b]+nums[c]+nums[d]<target)
        			    c++;
        			else if(nums[a]+nums[b]+nums[c]+nums[d]>target)
        			    d--;
        			else{
        				res.push_back({nums[a],nums[b],nums[c],nums[d]});
        				while(c<d&&nums[c+1]==nums[c])      //确保nums[c] 改变了
        				    c++;
        				while(c<d&&nums[d-1]==nums[d])      //确保nums[d] 改变了
        				    d--;
        				c++;
        				d--;
					}
				}
			}
		}
		return res;
    }
};