三数之和
2019/09/08
题目重述
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
思路
1.对数组进行排序。从向量的第一个元素开始,跳过开头重复元素,选中重复元素中最后一个固定。再使用左右指针指向nums[i]后面的两端,数字分别是 nums[L] 和 nums[R]。计算三个数的和sum,判断sum是否为0,若满足则添加进vector<vector<int>> result中。
2.若nums[i] > 0,则三数之和sum必然无法等于0,结束循环。
3.当sum == 0时,nums[L] == nums[L+1],则会导致结果重复,应跳过。
当sum == 0时,nums[R] == nums[R-1],则会导致结果重复,应跳过。
2019/09/07
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int target;
vector<vector<int>> ans;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); i++) {
if (i > 0 && nums[i] == nums[i - 1]) continue;
if ((target = nums[i]) > 0) break;
int l = i + 1, r = nums.size() - 1;
while (l < r) {
if (nums[l] + nums[r] + target < 0) ++l;
else if (nums[l] + nums[r] + target > 0) --r;
else {
ans.push_back({target, nums[l], nums[r]});
++l, --r;
while (l < r && nums[l] == nums[l - 1]) ++l;
while (l < r && nums[r] == nums[r + 1]) --r;
}
}
}
return ans;
}
};
选择“三数之和”的记录理由:在很多类似题目中都有很好的借鉴意义。
tip1:
1.如何打印二维vector数组;
vector<vector<int>> result; //result本应有值,此处仅给思路
for(int i = 0; i < result.size(); i++) {
for(int j = 0; j < result[i].size(); j++)
cout<<result[i][j]<<" ";
cout<<endl;
}
2.迭代器不能互相比较大小;
3.注意if (i > 0 && nums[i] == nums[i - 1])中的判断条件不能反(否则会先判断向量第一个元素与它之前的元素是否相等,但第一个元素没有前一个元素,程序会出现错误);