题解:
这是一道很好的算法题。它重点考验的是我们对于前后指针法的理解。
在本题中,我们首先要理清思路。
1.因为要求是三数之和,所以固定一个数不动,让另外两个数移动搜索求和为0。所以采用前后指针法。
2.因为题目要求不能有重复的数组,且基于第1点,所以要提前将数组排序。
3. 在进行搜索移动时,要注意跳过重复的部分。
基于以上几点,我们代码的总体思路是:
从头开始固定一个数字,让另外两个一前一后与固定的数字相加,直到两个数字相遇,然后固定数字向后移动。
如果和等于0,记录此次三个数。
如果和小于0,前头的向后移动。
如果和大于0,后头的向前移动。
代码如下:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(), nums.end());//排序
vector<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)
思考:如果用哈希表怎么解决,以及方不方便呢?
- 算法,就像一张菜谱。”——瓦辛·拉提夫(Waseem Latif)
如有错误,敬请斧正