leetcode之3Sum

原题如下：

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

• Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
• The solution set must not contain duplicate triplets.

    For example, given array S = {-1 0 1 2 -1 -4},

    A solution set is:
    (-1, 0, 1)
    (-1, -1, 2)

经过昨天的搜狗面试，我更加注意算法的时间复杂度问题，显然这道题存在暴力解法，其复杂度为O（n*n*n），现在考虑的是降低复杂度的问题，首先进行排序，其时间复杂度为O（nlogn），然后当为2Sum问题时可以采用左右指针查找的方法（这里因为数组中存在重复元素，所以需要去重），此时查找复杂度为O（n），所以总的时间复杂度为O（nlogn），在求解3Sum问题时，首先选中一个元素，然后在剩余的元素中进行2Sum的查找，此时总的时间复杂度为O（n*n），这里需要注意的仍是去重问题，首先选择第一个元素时需要去重，接下来在左右指针移动过程中仍要去重。

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
		sort(num.begin(),num.end());
		vector<vector<int>>vv;
		if(num.size() < 3)
			return vv;
		for(int i = 0; i < num.size() - 2; i++){
			if(i > 0 && num[i] == num[i - 1])
				continue;
			twoSum(vv,num,i + 1,num[i] * -1);
		}
		return vv;
    }
	void twoSum(vector<vector<int>>&vv,vector<int>num,int index,int target){
		int left = index,right = num.size() - 1;
		while(left < right){
			int temp = num[left] + num[right];
			if(temp == target ){
			    vector<int>triplet;
				triplet.push_back(target * -1);
				triplet.push_back(num[left]);
				triplet.push_back(num[right]);				
				vv.push_back(triplet);
				left++;
				right--;
				while ( left < num.size() && num[left] == num[left - 1] )
					left++;
				while( right >= 0 && num[right] == num[right + 1])
					right--;
			}
			else if(temp > target){
			    right--;
			}
			else
				left++;
		}
	}
};