LeetCode算法第15题 三数之和 Java

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

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

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

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

package algorithm;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.*;

public class 三数之和 {

	public static void main(String[] args) {
		int[] nums = {-1,0,1,2,-1,-4}; 
		
		//先给数组排序从小到大
		Arrays.sort(nums);
		//排序后是 -4 -1 -1 0 1 2
		
		//创建一个集合存放list
		List<List<Integer>> resurt = new ArrayList<>();
		
		for(int i=0;i<nums.length;i++) {
		
			//当i大于零时后面出现重复的数字就运用不上，
			//就要对i指针所指的数字进行查重排除数组中的重复的数字
			if(i>0&&nums[i]==nums[i-1])continue;
			
			//设置一个数字方便后面进行比较
			int sum = -nums[i];
			
			//现在此指针指向 -1
			int j = i+1;
			//现在此指针指向尾部 2
			int k = nums.length-1;
			
			//循环开始
			while(j<k) {
				//如果此两数相加等于sum就存储这三个数字到list集合
				//并进行j向前移，同时k向后移
				if(nums[j]+nums[k] == sum) {
					List<Integer> list = new ArrayList();
					list.add(nums[i]);
					list.add(nums[j]);
					list.add(nums[k]);
					resurt.add(list);
					j++;k--;
					
					//进行 j 和 k的查重并进行限制j最多移到尾部，k最多移到j的后一位
					while(j<nums.length&&nums[j]==nums[j-1])j++;
					while(k>j&&nums[k]==nums[k+1])k--;
				//当两数相加大于是k-- 使数字变小
				}else if(nums[j]+nums[k]>sum) {
					k--;
				//两数小的情况下 j++ 使数字变大
				}else {
					j++;
				}
			}
		}
		System.out.println(resurt);
	}
}