三数之和

三数之和题目

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。

个人理解

首先对数组进行排序，排序后固定一个数 nums[i]，再使用左右指针指向 nums[i]后面的两端，数字分别为 nums[j]和 nums[k]，计算三个数的和 sum 判断是否满足为 0，满足则添加进list集合。如果 nums[i]大于 0，则三数之和必然无法等于 0，结束循环。如果 nums[i] == nums[i-1]，则说明该数字重复，会导致结果重复，所以应该跳过。当 sum == 0 时，nums[j] == nums[j+1] 则会导致结果重复，应该跳过，j++,当 sum == 0 时，nums[k] == nums[k-1],则会导致结果重复，应该跳过，k–。
类似的题还有很多，都是先固定一方，然后双指针解法。

解题代码

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

public class TreeNums {
    /**
     * 三数之和
     * @param nums
     * @return
     */
    public static List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>>result=new ArrayList<>();
        if(nums==null||nums.length<3)
            return result;
        Arrays.sort(nums);
        for(int i=0;i<nums.length;i++){
            if(nums[i]>0)
                break;
            if(i>0&&nums[i]==nums[i-1])//去重
                continue;
            int j=i+1;
            int k=nums.length-1;
            while(j<k){
                int sum=nums[i]+nums[j]+nums[k];
                if(sum<0){
                    j++;
                }else if(sum>0){
                    k--;
                }else{
                    List<Integer>list=new ArrayList<>(3);
                    list.add(nums[i]);
                    list.add(nums[j]);
                    list.add(nums[k]);
                    result.add(list);
                    while(j<k&&nums[j]==nums[j+1]) j++;//去重
                    while(j<k&&nums[k]==nums[k-1])k--;//去重
                    j++;
                    k--;
                }
            }
        }
        return  result;
    }

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

运行结果