题目: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)
思路:
从前往后,遇到重复数字,往后走,一开始我卡在这里,我认为连续几个一起出现的情况不曾考虑,其实已经考虑,还有就是这样考虑的是为了比如说有2个1;1,2,3是一对,如果到下一个1不continue,那么后面就会继续有1,2,3.就会造成重复。
在后面出现的情况就是判断后面left和right的和的大小与target-nums[i]。这边容易,不再赘述。
代码:
class Solution { public: vector<vector<int> > threeSum(vector<int> &nums) { vector<vector<int> >result; int length=nums.size(); int count=0; sort(nums.begin(),nums.end()); for(int i=0;i<=length-3;i++){ if(i>0&&nums[i-1]==nums[i]){ continue; } //一开始误解程序意思,只要是后面有重复的,不考虑,因为那个第一个已经处理过 int left=i+1,right=length-1; while(left<right){ int sum=0-nums[i]; int cursum=nums[left]+nums[right]; if(cursum==sum){ vector<int>temp; temp.push_back(nums[i]); temp.push_back(nums[left]); temp.push_back(nums[right]); result.push_back(temp); left++;right--; while(nums[left]==nums[left-1]){ left++; } while(nums[right]==nums[right+1]){ right--; }//之前考虑了三个连在一起的情况,是这样的,如果两者连在一起,那就是正好插入进去,这个时候 //left++,right--,但不能够有重复的数字,所以这样足够,而我们所谓的出现三个数字相同的情况就是i,left,right是正好连在一起的了 } if(cursum<sum){ left++; } if(cursum>sum){ right--; } } } return result; } };