这个题使用哈希图来解非常的麻烦,因为要处理相等的情况。 使用双指针可以很好的解决这个问题。先规定第一个数,剩下两个数用双指针找出来。三个数中最左边的数先被确定下来,中间的数和右边的数可以移动,在处理之前先对数组进行排序,这样右边的数就大于中间的数,如果三个数之和大于0,则向内移动右边的数
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(),nums.end());
int n = nums.size();
vector<vector<int>> res;
for(int i = 0; i < n; i++)
{
if(nums[i] > 0)
break;
if(i > 0 && nums[i] == nums[i - 1])
continue;
int left = i + 1, right = n - 1;
while(left < right)
{
if(nums[i] + nums[left] + nums[right] == 0)
{
res.push_back({nums[i], nums[left], nums[right]});
while(left + 1 <= right && nums[left] == nums[left + 1])
left++;
while(right - 1 >= left && nums[right] == nums[right - 1])
right--;
left++; right--;
}
else if (nums[i] + nums[left] + nums[right] > 0)
{
while(right - 1 >= left && nums[right] == nums[right - 1])
right--;
right--;
}
else
{
while(left + 1 <= right && nums[left] == nums[left + 1])
left++;
left++;
}
}
}
return res;
}
};
|