算法练习专题——LeetCode系列之 ThreeSum

Question:

Given an array S of n integers, are there elements a, b, c in S 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.

假设现有一个拥有n个整数的数组S，要求找出所有不重复的数组中任意三个和为0的整数组合。

Example：

    For example, given array S = [-1, 0, 1, 2, -1, -4],

    A solution set is:
    [
      [-1, 0, 1],
      [-1, -1, 2]
    ]

    注意，[-1, 0, 1] 和[0, 1, -1]属于重复组合，结果中只能出现一个！

Answer Template:

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {

    }
}

---

Solution:

这个题目的解题关键在于，先将整个数组排序，这样相等的整数就会相邻，然后我们从左往右遍历num.length-2个元素，对于每一个元素，我们对剩下的元素进行一次双向的遍历，并过滤掉重复的元素。

    public static List<List<Integer>> threeSum(int[] nums) {
        //先将整个数组排序
        Arrays.sort(nums);
        List<List<Integer>> result = new ArrayList<>();
        for(int i=0; i<nums.length-2; i++) {
            //如果nums[i]与前一个元素相同，则跳过
            if(i==0 || (i>0&&nums[i]!=nums[i-1])) {
                //li为左向遍历的索引，ri为右向遍历的索引，sum为num[i]的相反数
                int li=i+1, ri=nums.length-1, sum = 0-nums[i];
                //只要两个索引没有相交
                while(li<ri) {
                    //两数相加等于sum
                    if(nums[li]+nums[ri] == sum) {
                        //在result中增加这三个数的组合
                        result.add(Arrays.asList(nums[i],nums[li],nums[ri]));
                        //继续遍历，跳过重复元素
                        while(li<ri && nums[li]==nums[li+1]) li++;
                        while(li<ri && nums[ri]==nums[ri-1]) ri--;
                        li++;
                        ri--;
                    }
                    //如果两数相加小于sum，左索引+1，去找更大的数
                    else if(nums[li] +nums[ri] <sum) {
                        //继续遍历，跳过重复元素
                        while(li<ri && nums[li]==nums[li+1]) li++;
                        li++;
                    }
                    //如果两数相加大于sum，又索引-1，去找更小的数
                    else {
                        //继续遍历，跳过重复元素
                        while(li<ri && nums[ri]==nums[ri-1]) ri--;
                        ri--;
                    }
                }
            }
        }
        return result;
    }

本节博客的源码全部放在这里，欢迎大家下载，下载后记得修改源码中的package名字哦。