15.ThreeSum

问题

Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

The solution set must not contain duplicate triplets.

Example:

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

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

解决方案

public List<List<Integer>> threeSum(int[] nums) {
	List<List<Integer>> result = new LinkedList<>();
	
	if (nums != null &&nums.length >2) {
		
		//首先对数组进行排序
		Arrays.sort(nums);
		//我们假设取第 i 个数为结果
		for(int i= 0;i<nums.length-2;)
		{
			//第二个数的可能起始位置
			int j = i+1;
			//第三个数的可能起始位置
			int k = nums.length - 1;
			
			while (j < k) {
				if (nums[j] + nums[k] == -nums[i]) {
					//当我们找到满足条件的解时，我们将结果添加到结果集里
					List<Integer> list = new ArrayList<>(3);
					
					list.add(nums[i]);
					list.add(nums[j]);
					list.add(nums[k]);
					result.add(list);
					//移动位置寻找下一组解
					k--;
					j++;
					
					//下面两个while是用来判断是否越界和判断移动后的数是否和之前的数相同，若相同则跳过该数
					while (j<k && nums[j] == nums[j-1]) {
						j++;
					}
					while (j<k&&nums[k] == nums[k+1]) {
						k--;
					}
				}
				//和大于 0，让k向前移动到小一点的数
				else if (nums[j]+nums[k]>-nums[i]) {
					k--;
					//下面是用来判断是否越界和判断移动后的数是否和之前的数相同，若相同则跳过该数
					while (j<k&&nums[k] == nums[k+1]) {
						k--;
					}
				}
				//和小于 0，让j向后移动到大一点的数
				else {
					j++;
					//下面是用来判断是否越界和判断移动后的数是否和之前的数相同，若相同则跳过该数
					while (j<k&&nums[j] == nums[j-1]) {
						j++;
					}
				}
			}
			//指向下一个要处理的数
			i++;
			//从左向右找第一个与之前处理的数的不同的数的下标
			while (i<nums.length-2&&nums[i] == nums[i-1]) {
				i++;
			}
		}
	}
	  
	  return result;
        
   }