解题思路

  • 暴力法搜索为 O(N^3) 时间复杂度,可通过双指针动态消去无效解来优化效率。
  • 双指针法铺垫: 先将给定 nums 排序,复杂度为 O(NlogN)
  • 双指针法思路: 固定 3 个指针中最左(最小)数字的指针 k,双指针 i,j 分设在数组索引 (k, len(nums)) 两端,通过双指针交替向中间移动,记录对于每个固定指针 k 的所有满足 nums[k] + nums[i] + nums[j] == 0 的 i,j 组合:
  1. 当 nums[k] > 0 时直接break跳出:因为 nums[j] >= nums[i] >= nums[k] > 0,即 3 个数字都大于 00 ,在此固定指针 k 之后不可能再找到结果了。
  2. 当 k > 0 且 nums[k] == nums[k - 1]时即跳过此元素nums[k]:因为已经将 nums[k - 1] 的所有组合加入到结果中,本次双指针搜索只会得到重复组合。
  3. i,j 分设在数组索引 (k, len(nums)) 两端,当i < j时循环计算s = nums[k] + nums[i] + nums[j],并按照以下规则执行双指针移动:
    • 当s < 0时,i += 1并跳过所有重复的nums[i];
    • 当s > 0时,j -= 1并跳过所有重复的nums[j];
    • 当s == 0时,记录组合[k, i, j]至res,执行i += 1和j -= 1并跳过所有重复的nums[i]和nums[j],防止记录到重复组合

图解

Video_2020-03-25_210551.wmv (146.72KB)

复杂度分析

时间复杂度 O(N^2):其中固定指针k循环复杂度 O(N),双指针 i,j 复杂度 O(N)。
空间复杂度 O(1):指针使用常数大小的额外空间。


代码

  1. // JAVA
  2. class Solution {
  3. public List<List<Integer>> threeSum(int[] nums) {
  4. Arrays.sort(nums);
  5. List<List<Integer>> res = new ArrayList<>();
  6. for(int i = 0; i < nums.length - 2; i++){
  7. // nums[k]为非负数,就不能满足a+b+c=0了
  8. if(nums[i] > 0) break;
  9. // 跳过计算过的数据,同时防止结果重复
  10. if(i > 0 && nums[i] == nums[i - 1]) {
  11. continue;
  12. }
  13. int leftPoint = i + 1, rightPoint = nums.length - 1;
  14. while(leftPoint < rightPoint){
  15. int sum = nums[i] + nums[leftPoint] + nums[rightPoint];
  16. if(sum < 0){
  17. leftPoint++;
  18. } else if (sum > 0) {
  19. rightPoint--;
  20. } else {
  21. res.add(Arrays.asList(nums[i], nums[leftPoint], nums[rightPoint]));
  22. while (leftPoint < rightPoint && nums[leftPoint] == nums[leftPoint+1]) {
  23. leftPoint++;
  24. }
  25. while(leftPoint < rightPoint && nums[rightPoint] == nums[rightPoint-1]) {
  26. rightPoint--;
  27. }
  28. leftPoint++;
  29. rightPoint--;
  30. }
  31. }
  32. }
  33. return res;
  34. }
  35. }
  1. # python
  2. class Solution:
  3. def threeSum(self, nums: List[int]) -> List[List[int]]:
  4. res = []
  5. nums.sort()
  6. length = len(nums)
  7. if (not nums or length<3):
  8. return []
  9. for i in range(length):
  10. if nums[i] > 0:
  11. break
  12. if i > 0 and nums[i] == nums[i-1]:
  13. continue
  14. leftPoint, rightPoint = i + 1, length - 1
  15. while leftPoint < rightPoint:
  16. total = nums[i] + nums[leftPoint] + nums[rightPoint]
  17. if total < 0:
  18. leftPoint += 1
  19. elif total > 0:
  20. rightPoint -= 1
  21. else:
  22. res.append([nums[i], nums[leftPoint], nums[rightPoint]])
  23. while leftPoint < rightPoint and nums[leftPoint] == nums[leftPoint+1]:
  24. leftPoint += 1
  25. while leftPoint < rightPoint and nums[rightPoint] == nums[rightPoint-1]:
  26. rightPoint -= 1
  27. leftPoint += 1
  28. rightPoint -= 1
  29. return res
  1. // C++
  2. class Solution {
  3. public:
  4. vector<vector<int>> threeSum(vector<int>& nums) {
  5. vector<vector<int>> res;
  6. vector<int > vtemp;
  7. sort(nums.begin(), nums.end()); // sort
  8. for (int i = 0; i < nums.size(); i++) {
  9. if (nums[i] > 0) {
  10. break;
  11. }
  12. if (i > 0 && nums[i] == nums[i-1]) {
  13. continue;
  14. }
  15. int leftPoint = i + 1;
  16. int rightPoint = nums.size() - 1;
  17. while (leftPoint < rightPoint) {
  18. int sum = nums[leftPoint] + nums[rightPoint] + nums[i];
  19. if (sum < 0) {
  20. leftPoint++;
  21. } else if(sum > 0) {
  22. rightPoint--;
  23. } else{
  24. vector<int > vtemp{nums[i], nums[leftPoint], nums[rightPoint]};
  25. res.push_back(vtemp);
  26. vtemp.clear();
  27. while (leftPoint < rightPoint && nums[leftPoint] == nums[leftPoint+1]) {
  28. leftPoint++;
  29. }
  30. while (leftPoint < rightPoint && nums[rightPoint] == nums[rightPoint-1]) {
  31. rightPoint--;
  32. }
  33. leftPoint++;
  34. rightPoint--;
  35. }
  36. }
  37. }
  38. return res;
  39. }
  40. };