Leet Code 15 3Sum - 三个数的和等于0 - Java

问题原始链接 https://leetcode.com/problems/3sum

给定一个数组，找到数组中所有的和等于0的三个元素组合。结果集中不能包含重复的三元组。

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

结果集：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

public class Solution {
  public static List<List<Integer>> threeSum(int[] nums) {
    List<List<Integer>> result = new ArrayList<List<Integer>>();
    if (nums == null) {
      return result;
    }
    Arrays.sort(nums);
    for (int i = 0; i < nums.length - 2; i++) {
      if (i == 0 || nums[i] > nums[i - 1]) {
        int j = i + 1;
        int k = nums.length - 1;
        while (j < k) {
          if (nums[i] + nums[j] + nums[k] == 0) {
            result.add(Arrays.asList(nums[i], nums[j], nums[k]));
            j++;
            k--;

            while (j < k && nums[j] == nums[j - 1]) {
              j++;
            }
            while (j < k && nums[k] == nums[k + 1]) {
              k--;
            }
          } else if (nums[i] + nums[j] + nums[k] < 0) {
            j++;
          } else {
            k--;
          }
        }
      }
    }
    return result;
  }
}

 

转载于:https://my.oschina.net/mistymarch/blog/687913