给定一个数组,找出3个数相加为0的所有组合。
例如:给出0,1,-1,4,5,3,-7
组合为0,-1, 1;3, 4,-7
解题思路:这里要求a+b+c=0,只需要满足a+b=-c即可。首先将数组按递增排序,通过遍历,当前数若大于0即可跳出循环。需要注意的是,如果当前数与上一个数相等,且当前数大于0,那么不用再继续判断,只需要直接跳到下一个数即可。
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
sort(nums.begin(), nums.end());
for (int k = 0; k < nums.size(); ++k) {
if (nums[k] > 0) break;
if (k > 0 && nums[k] == nums[k - 1]) continue; //当前数与上一个数相等时,可以直接跳到下一个数,避免重复统计
int target = 0 - nums[k];
int i = k + 1, j = nums.size() - 1;
while (i < j) {
if (nums[i] + nums[j] == target) {
res.push_back({nums[k], nums[i], nums[j]});
while (i < j && nums[i] == nums[i + 1]) ++i;
while (i < j && nums[j] == nums[j - 1]) --j;
++i; --j;
} else if (nums[i] + nums[j] < target) ++i;
else --j;
}
}
return res;
}