1. 程式人生 > >LeetCode18:四數之和(Medium)

LeetCode18:四數之和(Medium)

題目描述:

給定一個包含 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]
]

這個跟三數之和的思維一樣的,就是多添加了個for迴圈,用了四個指標

c++實現:

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        vector<int> sum(4);
        vector<vector<int> > res;
        int len=nums.size();
        int tsum;
        int i,j,l,k;
        for (i=0;i<len-3;i++)
        {
        	if(i>0&&nums[i]==nums[i-1])
        		continue;
        	for(j=i+1;j<len-2;j++){
        		if(j>i+1 && nums[j]==nums[j-1])
        			continue;
        		for(l=j+1,k=len-1;l<k;){
        			tsum=nums[i]+nums[j]+nums[l]+nums[k];
        			if(tsum==target){
        				sum[0]=nums[i];
        				sum[1]=nums[j];
        				sum[2]=nums[l];
        				sum[3]=nums[k];
        				res.push_back(sum);
        				while(l<k && nums[l]==nums[l+1])
        					l++;
        				while(l<k && nums[k]==nums[k-1])
        					k--;
        				l++;
        				k--;
					}
					else if (tsum<target)
						l++;
					else
						k--;
				}
			}
		}
		return res;
    }
};