双指针系列-Leetcode 四数之和

问题描述

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

解题报告

略。

实现代码

class Solution{
	public: 
	vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        vector<vector<int> > ans;
        if(nums.size()<4)
            return ans;
        int p1,p2,p3,p4,n=nums.size(),sum;
        for(p1=0;p1<n-3;p1++){
            if(nums[p1]>target&&nums[p1]>=0) break;
        	if(p1>0&&nums[p1]==nums[p1-1]) continue;   
        	for(p2=p1+1;p2<n-2;p2++){
                if(nums[p1]+nums[p2]>target&&nums[p2]>=0) break;
        		if(p2>p1+1&&nums[p2]==nums[p2-1])continue;
        		p3=p2+1,p4=n-1;
        		while(p3<p4){
                    sum=nums[p1]+nums[p2]+nums[p3]+nums[p4];
        			if(sum<target)while(p3<p4&&nums[p3] == nums[++p3]);
        			else if(sum>target)while(p3<p4&& nums[p4] == nums[--p4]);
        			else{
        				ans.push_back({nums[p1],nums[p2],nums[p3],nums[p4]});
        				while(p3<p4&&nums[p3]==nums[++p3]);
        				while(p3<p4&&nums[p4]==nums[--p4]);
					}
				}
			}
		}
		return ans;
    }
};

参考资料

[1] Leetcode 四数之和