题目出处
给定一个数组,查找所有和等于0的三个数。
如: 给定的数组为:[-1, 0, 1, 2, -1, -4]
返回的结果是:[ [-1, -1, 2], [-1, 0, 1]]
分析
可以采用基本算法来解决:
先对数组进行排序
定义三个指针: i, j, k
i: 从排序后的数组开始移动。
k: 从数组的最后往前移动。
j: 在[i, k]之间移动
如果i, j, k对应的数之和为sum;
如果sum== 0, 得到一个解, 同时移动j, k。
如果sum > 0: 移动k
如果sum < 0: 移动j
如果 j > k: 则移动i, 并重新定义 k, j.
同时对算法作两点优化来减少比较的次数
k :没必要每次都从最后元素开始移动,所以需要记录k的最后位置。
i: 没必要移动最后, 要以i > 0 为终点。
解法二
还有一种比较快的办法,就是利用hash表。估算了一下,算法复杂度都在o(n*n), 差不多, 就没实现了。
代码
vector<vector<int>> threeSum(vector<int>& nums) { if(nums.size() < 3) return vector<vector<int>>(); sort(nums.begin(), nums.end()); if(nums.front() > 0 || nums.back() < 0) return vector<vector<int>>(); int last = nums.size()-1; vector<vector<int>> ret; for(int i = 0; nums[i] <= 0; i++){ if(i > 0 && nums[i] == nums[i-1]) continue; int j = i + 1; int k = last; while(j < k){ if(j > i+1 && nums[j] == nums[j-1]) {j += 1;continue;} int num = nums[i] + nums[j] + nums[k]; if(num < 0){j += 1;} else if(num > 0){ k -= 1; if(j == i + 1) last = k; } else{ ret.push_back({nums[i], nums[j], nums[k]}); j += 1; k += 1; } } } return ret; }