【题目描述】
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
【解答】
按元素枚举(保证每一次枚举的元素的下一个都比当前值大)
- class Solution {
- public:
- vector<vector<int>> threeSum(vector<int>& nums) {
- int n = nums.size();
- sort(nums.begin(), nums.end());
- vector<vector<int>> ans;
- int a=0,b=0,c=0;
- for (a = 0; a < n; ++a) {
-
- if (a > 0 && nums[a] == nums[a - 1]) {
- continue;
- }
-
- c=n-1;
- int target = -nums[a];
-
- for (b = a + 1; b < n; ++b) {
-
- if (b > a + 1 && nums[b] == nums[b - 1]) {
- continue;
- }
-
- while (b < c && nums[b] + nums[c] > target) {
- --c;
- }
-
- if (b >= c) {
- break;
- }
- if (nums[b] + nums[c] == target) {
- ans.push_back({nums[a], nums[b], nums[c]});
- }
- }
- }
- return ans;
- }
- };
(题解的结构写的很清晰了:
链接:https://leetcode.cn/problems/3sum/solution/san-shu-zhi-he-by-leetcode-solution/)