三数之和等于零的组合排列
LeetCode 三数之和 (中级算法1)
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
思路(双指针)
三个数字的类型有两种可能:
- 正负相加
- 全零
对于第一种可能性,需要去重:
- 在枚举第一个数字a时,保证不重复
- 在找到第二第三个数字后去重
在有序数组里,去重就是将指针指向下一位不重复的数字;因为排序后相同元素在一起,用nums.at(i) == nums.at(i+1)
来判断即可
几个要点:
- a只能小于等于零
- 双指针遍历时需要检查是否越界
- 去重后双指针再后移一位
- 先
push_back(temp_vector)
再对第二第三个进行去重
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
// sort the array
sort(nums.begin(), nums.end());
vector<vector<int>> res;
int n = nums.size();
// 枚举第一个元素,双指针遍历
for (int i = 0; i < n - 2; ++i){
if (nums[i] > 0) break;
// 第一个元素去重
if (i > 0 && nums[i] == nums[i-1]) continue;
// 双指针
int left = i + 1;
int right = n - 1;
// 记住要检查是否越界:left < right
while (left < right){
int sum = nums[i] + nums[left] + nums[right];
// 找到第二第三个元素后
if (sum > 0){
--right; // 继续向后找
} else if (sum < 0){
++left; // 继续向前找
} else {
res.push_back({nums[i], nums[left], nums[right]});
// 第二个元素去重
while(left < right && nums[left] == nums[++left]);
while(left < right && nums[right] == nums[--right]);
}
}
}
return res;
}
};
问题
nums[left] == nums[left++]; nums[left] == nums[++left]
哪个永真?- 为什么把
n
换成nums.size()
就会报错undefined behavior
? - 先检查
sum < 0
效率比先检查sum > 0
低 - 为什么第一个if条件为
sum == 0
时会超出时间限制?