雙指針系列-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 四數之和