
这是一道很好的算法题。它重点考验的是我们对于前后指针法的理解。
在本题中,我们首先要理清思路。
1.因为要求是三数之和,所以固定一个数不动,让另外两个数移动搜索求和为0。所以采用前后指针法。
2.因为题目要求不能有重复的数组,且基于第1点,所以要提前将数组排序。
3. 在进行搜索移动时,要注意跳过重复的部分。
基于以上几点,我们代码的总体思路是:
从头开始固定一个数字,让另外两个一前一后与固定的数字相加,直到两个数字相遇,然后固定数字向后移动。
如果和等于0,记录此次三个数。
如果和小于0,前头的向后移动。
如果和大于0,后头的向前移动。
代码如下:
- class Solution {
- public:
- vector
int>> threeSum(vector<int>& nums) { - sort(nums.begin(), nums.end());//排序
- vector
int>> ret; - if(nums.size() < 3) return ret;//如果数组长度小于3,直接返回
- for(int i = 0; i < nums.size() - 2; i++)//固定数字从头开始,直到倒数第2位结束
- {
- int left = i + 1, right = nums.size() - 1;//前后指针
- while(left < right)
- {
- int sum = nums[i] + nums[left] + nums[right];//求和
-
- if(sum == 0)
- {
- ret.push_back({nums[i], nums[left], nums[right]});//记录数组
-
- while(left < right &&
- (nums[left] == nums[left++ + 1] && nums[right] == nums[right-- -1]))
- { }//保证不相遇下,让前指针或后指针其中一个走到不同数字的下标
- }
-
- else if(sum < 0)
- {
- while(left < right && nums[left] == nums[left++ + 1]){}
- //让前指针走到不同数字的下标
- }
-
- else
- {
- while(left < right && nums[right] == nums[right-- - 1]){}
- //让后指针走到不同数字的下标
- }
-
- }
- while(i < nums.size() - 2 && nums[i] == nums[i + 1]) i++;
- //让固定数字走到不同数字的下标,与前后指针一样,防止重复
- }
- return ret;
- }
- };
时间复杂度:O(n^2)
空间复杂度:O(n)
思考:如果用哈希表怎么解决,以及方不方便呢?
如有错误,敬请斧正