题目链接:https://leetcode.com/problems/3sum/#/descriptionGiven 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: 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]
]
思路:(用递归超时)
先升序排序,然后用第一重for循环确定第一个数字。
然后在第二重循环里,第二、第三个数字分别从两端往中间扫。
如果三个数的sum等于0,得到一组解。
如果三个数的sum小于0,说明需要增大,所以第二个数往右移。
如果三个数的sum大于0,说明需要减小,所以第三个数往左移。
时间复杂度:O(n2)
class Solution { public: vector<vector<int>> threeSum(vector<int>& nums) { vector<vector<int>> res; int len=nums.size(); if(len<3){ return res; } sort(nums.begin(),nums.end()); for(int i=0;i<len;i++) { if(nums[i]>0)break; if(i>0 && nums[i]==nums[i-1])continue; int begin=i+1,end=len-1; while(begin<end) { int sum=nums[i]+nums[begin]+nums[end]; if(sum==0){ vector<int> t; t.push_back(nums[i]); t.push_back(nums[begin]); t.push_back(nums[end]); res.push_back(t); begin++;end--; while(begin<end && nums[begin]==nums[begin-1])begin++; while(begin<end && nums[end]==nums[end+1])end--; } else if(sum>0) { end--; } else begin++; } } return res; } };