代码随想录-026-LeetCode15.三数之和

前言

“代码随想录”刷题记录。总结笔记均会放在“算法刷题-代码随想录”该专栏下，以下为原文的链接。
代码随想录此题链接

题目

给你一个包含 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. 双指针法

思路（定义变量）

1. 先将数组从小到大排序，然后使用三个指针指向数组的不同位置。

• start作为起始指针。
• left和right分别为起始指针后面区域的头尾指针。

2. result作为返回的vector<vector< int>> （List<List< int>>）

2. 本题思路分析（较为复杂，直接看卡哥的解析）：

将数组按从小到大的顺序排序，排完序后，start作为起始指针，left在start之后，right在末尾。
外层一个for循环让start从开始循环到倒数第三个数也就是start<=nums.length-3（一共两层循环，这个是外层循环），内层循环是left在start之后，right在末尾；内层循环每次循环时，若三者大于0，将right–；若三者之和小于0，则将left++，若三者之和等于0，此时将结果加入result中，,并将两个指针收缩，即left++，right–。

拿这个nums数组来举例，首先将数组排序，然后有一层for循环，i从下标0的地方开始，同时定一个下标left 定义在i+1的位置上，定义下标right 在数组结尾的位置上。

依然还是在数组中找到 abc 使得a + b +c =0，我们这里相当于 a = nums[i]，b = nums[left]，c = nums[right]。

接下来如何移动left 和right呢， 如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了，因为数组是排序后了，所以right下标就应该向左移动，这样才能让三数之和小一些。

如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了，left 就向右移动，才能让三数之和大一些，直到left与right相遇为止。

3. 算法实现

• 代码：
  JAVA

public List<List<Integer>> threeSum(int[] nums) {
 List<List<Integer>> result = new ArrayList<List<Integer>>();
 Arrays.sort(nums);        
 for(int start = 0;start <= nums.length - 3;start++){
     //剪枝
     if(nums[start] > 0){
         return result;
     }
     //去重
     if(start > 0 && nums[start] == nums[start - 1]){
         continue;
     }
     int left = start + 1,right = nums.length - 1;
     while(left < right){
         int target = nums[start] + nums[left] + nums[right];
         if(target > 0){
             right--;
         }else if(target < 0){
             left++;
         }else{
             List<Integer> res = new ArrayList<Integer>();
             Collections.addAll(res,nums[start],nums[left],nums[right]);
             result.add(res);
             //去重
             while(left < right && nums[left] == nums[left + 1]){
                 left++;
             }
             while(left < right && nums[right] == nums[right-1]){
                 right--;
             }
             left++;
             right--;
		   }
        }            
    }
    return result;
}

C++

#include <stdio.h> 
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
vector<vector<int>> threeSum(vector<int>& nums) {
	//双指针法
	vector<vector<int>>	result;
	//排除特例
	if(nums.size() < 3){
		return result;
	}
	//记得要排序
	sort(nums.begin(),nums.end());//引入algorithm 
	for(int i = 0,left = i + 1,right = nums.size() - 1;i < nums.size() - 2;i++,left = i + 1,right = nums.size() - 1){
		//排序后排除特例
		if(nums[i] > 0) {
			return result;
		}
		//**先去 i 的重** 
		if(i > 0 && nums[i] == nums[i - 1]){
			continue;
		}
		while(left < right){
			if(nums[i] + nums[left] + nums[right] > 0){//太大了 
				right--;
			}else if(nums[i] + nums[left] + nums[right] < 0){//太小了 
				left++;
			}else{
				vector<int> v = {nums[i],nums[left],nums[right]};
				result.push_back(v);
				//去重
				while(left < right && nums[left] == nums[left + 1]) left++;
				while(left < right && nums[right] == nums[right - 1]) right--;
				left++;
				right--;
			}
		}
	}
	return result;
}

4. 算法复杂度

1. 时间复杂度：O(n^2)
2. 空间复杂度：O(1)

5. 算法坑点

1. 此题题目中对于三个指针的去重问题。

对于起始指针i的去重问题

//先去 i 的重
if(i > 0 && nums[i] == nums[i - 1]){
continue;
}

对于left、right指针的去重问题

while(left < right && nums[left] == nums[left + 1]) left++;
while(left < right && nums[right] == nums[right - 1]) right–;

2. 此题目对于特殊情况的处理

//排除特例
if(nums.size() < 3){
return result;
}

//排序后循环中排除特例
if(nums[i] > 0) {
return result;
}

2. 内层循环写法问题：
   内层（2层）循环中，应该只有当nums[left]+nums[right]==target时，left和right才要同时收缩，所以2层循环中left++和right–不要写在while循环之外。
   正确代码片段如下：

while(left < right){
	if(nums[i] + nums[left] + nums[right] > 0){//太大了 
		right--;
	}else if(nums[i] + nums[left] + nums[right] < 0){//太小了 
		left++;
	}else{
		vector<int> v = {nums[i],nums[left],nums[right]};
		result.push_back(v);
		//去重
		while(left < right && nums[left] == nums[left + 1]) left++;
		while(left < right && nums[right] == nums[right - 1]) right--;
		left++;
		right--;
	}
}

错误代码片段如下：

//❌错误写法
for(int left = i + 1,right = nums.size() - 1;left < right;left++,right--){
    if(nums[left] + nums[right] > target){
        right--;
    }else if(nums[left] + nums[right] < target){
        left++;
    }else{
        vector<int> tmp = {nums[i],nums[left],nums[right]};
        result.push_back(tmp);
        //去重
        while(left < right && nums[left] == nums[left + 1]) left++;
        while(left < right && nums[right] == nums[right - 1]) right--;
    }
    left++;
	right--;
}

2.java语法问题

• 对于数组的排序（以及对于集合的排序）
  使用Arrays工具类中的sort函数;如下

int[] arr = {3,4,1,2};
Arrays.sort(arr);//按照从小到大的顺序排序

- 使用Collections.addAll给ArrayList加入元素

ArrayList<Integer> arr1 = new ArrayList<Integer>();
Collections.addAll(arr1,1,2,3);//将1,2,3加入到arr1中