|
| 1 | +// class Solution { |
| 2 | +// public: |
| 3 | +// vector<vector<int>> threeSum(vector<int>& nums) { |
| 4 | +// set<vector<int>> ans; |
| 5 | +// map<int, int> counter; // {num : cnt} |
| 6 | +// vector<int> targets; |
| 7 | +// int target, left, right; |
| 8 | + |
| 9 | +// for(int i = 0; i < nums.size(); i++) |
| 10 | +// counter[nums[i]]++; |
| 11 | + |
| 12 | +// for(auto iter : counter) |
| 13 | +// targets.push_back(iter.first); |
| 14 | + |
| 15 | +// for(int i = 0; i < targets.size(); i++) { |
| 16 | +// target = -1 * targets[i]; |
| 17 | + |
| 18 | +// vector<int> new_nums; |
| 19 | +// counter[targets[i]]--; |
| 20 | +// for(auto iter : counter) { |
| 21 | +// for(int j = 0; j < iter.second; j++) |
| 22 | +// new_nums.push_back(iter.first); |
| 23 | +// } |
| 24 | +// left = 0; right = new_nums.size() - 1; |
| 25 | +// while(left < right) { |
| 26 | +// int sum = new_nums[left] + new_nums[right]; |
| 27 | +// if(sum == target && targets[i] <= new_nums[left] && new_nums[left] <= new_nums[right]) |
| 28 | +// ans.insert(vector<int>({targets[i], new_nums[left], new_nums[right]})); |
| 29 | + |
| 30 | +// if(sum < target) |
| 31 | +// left++; |
| 32 | +// else |
| 33 | +// right--; |
| 34 | +// } |
| 35 | +// counter[targets[i]]++; |
| 36 | +// } |
| 37 | + |
| 38 | +// return vector<vector<int>> (ans.begin(), ans.end()); |
| 39 | +// } |
| 40 | +// }; |
| 41 | + |
| 42 | +class Solution { |
| 43 | +public: |
| 44 | + vector<vector<int>> threeSum(vector<int>& nums) { |
| 45 | + vector<vector<int>> ans; |
| 46 | + int left, right, sum; |
| 47 | + |
| 48 | + sort(nums.begin(), nums.end()); |
| 49 | + |
| 50 | + for(int i = 0; i < nums.size(); i++) { |
| 51 | + if(i > 0 && nums[i] == nums[i - 1]) |
| 52 | + continue; |
| 53 | + |
| 54 | + left = i + 1; right = nums.size() - 1; |
| 55 | + while(left < right) { |
| 56 | + sum = nums[i] + nums[left] + nums[right]; |
| 57 | + if(sum == 0) { |
| 58 | + ans.push_back(vector<int> ({nums[i], nums[left], nums[right]})); |
| 59 | + left++; right--; |
| 60 | + while(left < right && nums[left] == nums[left - 1]) |
| 61 | + left++; |
| 62 | + while(left < right && nums[right] == nums[right + 1]) |
| 63 | + right--; |
| 64 | + } |
| 65 | + else if(sum < 0) |
| 66 | + left++; |
| 67 | + else if(sum > 0) |
| 68 | + right--; |
| 69 | + } |
| 70 | + } |
| 71 | + |
| 72 | + return ans; |
| 73 | + } |
| 74 | +}; |
| 75 | + |
0 commit comments