LeetCode题目：三数之和 思路及代码实现

题目：15. 给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

---

思路：三数相加等于零等同于两个数的和等于另一个数的相反数。先将数组排序，方便遍历，然后定义三个指针，sum、left、right，sum指向第一个元素、left指向sum后面的第一个元素、right指向数组的最后一个元素，如果left和right指向的元素之和小于sum指向的元素和的相反数，left右移，如果left和right指向的元素之和大于sum指向的元素和的相反数，right左移，如果相等，则三个指针指向的元素之和为0，这时还要同时将left和right往中间移动，寻找是否有其他两个数字相加等于sum指向的元素。右移sum，重复上面的过程。

Java代码：

package code;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Demo15 {
    static List<List<Integer>> threeSum(int[] nums){
        List<List<Integer>> result = new ArrayList<>();
        if (nums == null || nums.length<3)
            return result;
        List<Integer> singleList = new ArrayList<>();
        int sum = 0;//三个数相加等于0就代表两个数相加等于另一个数的相反数
        int left;
        int right; //left和right指向的元素相加等于sum的相反数
        int len = nums.length;
        //为了避免重复查找，首先对数组进行排序
        Arrays.sort(nums);
        while (sum <= len -3 && nums[sum]<=0){//最小的那个数都大于0，加上后面的两个数，肯定不等于0
            left = sum+1;
            right = len -1;
            while (left < right){
                if (-nums[sum] == nums[left] + nums[right]){
                    singleList.add(nums[sum]);
                    singleList.add(nums[left]);
                    singleList.add(nums[right]);
                    result.add(singleList);
                    singleList = new ArrayList<>();
                    while (left < right && nums[left+1] == nums[left])//去重
                        left++;
                    while (left < right && nums[right-1] == nums[right])
                        right--;
                    left++;
                    right--;
                }else if (-nums[sum] > nums[left] + nums[right]){
                    while (left < right && nums[left+1] == nums[left])
                        left++;
                    left++;
                }else {
                    while (left < right && nums[right-1] == nums[right])
                        right--;
                    right--;
                }
            }
            while (sum <= len -3 && nums[sum+1] == nums[sum])//这里要带上sum <= len -3 的条件 否则像{0,0,0}这样的数组会数组越界
                sum++;
            sum++;
        }
        return  result;
    }

    public static void main(String[] args) {
        int[] arr  = {0,0,0};
        threeSum(arr);
        System.out.println(threeSum(arr).toString());
    }
}