地址 https://leetcode-cn.com/problems/3sum/
给你一个包含 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
解答 1
两数之和的升级版本吧
首先排序数组
将所有数的出现次数使用哈希表记录
然后两重遍历 寻找第三个数字的时候 从哈希表中寻找
保证三种遍历 i j k 三个数由小到大排列 避免出现重复的答案
同是i保证每个数字之尝试一次 避免出现尝试重复的数字组合
class Solution { public: unordered_map<int, int> mm; vector<vector<int>> ans; vector<vector<int>> threeSum(vector<int>& nums) { sort(nums.begin(), nums.end()); for (auto& e : nums) { mm[e]++; } int prevI = 100000 + 10; int prevJ = 100000 + 10; for (int i = 0; i < nums.size(); i++) { if (nums[i] == prevI) continue; mm[nums[i]]--; for (int j = i + 1; j < nums.size(); j++) { if (nums[j] == prevJ) continue; mm[nums[j]]--; int find = 0 - nums[i] - nums[j]; if (find >= nums[i] && find >= nums[j]) { if ( mm[find] > 0) { ans.push_back(vector<int>{nums[i], nums[j], find}); } } mm[nums[j]]++; prevJ = nums[j]; } mm[nums[i]]++; prevI = nums[i]; } return ans; } };
复杂度是O(n2) 但是效率较低还有优化的空间
解答 2
复杂度同样是O(n2)
排序 一次选择某一个数作为第一个数 nums[i]
然后使用双指针求得另外两个数字,双指针范围(i+1 , nums.size()-1)
class Solution { public: unordered_map<int, int> mm; vector<vector<int>> ans; vector<vector<int>> threeSum(vector<int>& nums) { sort(nums.begin(), nums.end()); int prevI = 100010;int prevJ = 100010; for (int i = 0; i < nums.size(); i++) { if (nums[i] == prevI) continue; int l = i + 1; int r = nums.size() - 1; while (l < r) { if (nums[i] + nums[l] + nums[r] == 0) { ans.push_back(vector<int>{nums[i], nums[l], nums[r]}); l++; } else if (nums[i] + nums[l] + nums[r] > 0) { r--; } else if (nums[i] + nums[l] + nums[r] < 0) { l++; } } prevI = nums[i]; } return ans; } };
讲解 视频的时候 突然发现代码无法AC很尴尬
针对样例 升级了下代码
如下
class Solution { public: unordered_map<int, int> mm; vector<vector<int>> ans; vector<vector<int>> threeSum(vector<int>& nums) { //排序数组 避免重复答案 和双指针的预处理 sort(nums.begin(), nums.end()); int prevI = 100010;int prevJ = 100010; for (int i = 0; i < nums.size(); i++) { //避免选择则重读的数当做是第一个数 if (nums[i] == prevI) continue; int l = i + 1; int r = nums.size() - 1; //双指针选择后面两个数 while (l < r) { //根据两数之和的大小 来决定移动左指针或者右指针 if (nums[i] + nums[l] + nums[r] == 0) { ans.push_back(vector<int>{nums[i], nums[l], nums[r]}); l++; while(l <r && nums[l] == nums[l-1]) l++; } else if (nums[i] + nums[l] + nums[r] > 0) { r--; while(l <r && nums[r] == nums[r+1]) r--; } else if (nums[i] + nums[l] + nums[r] < 0) { l++; while(l <r && nums[l] == nums[l-1]) l++; } } prevI = nums[i]; } return ans; } };