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