算法练习专题——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名字哦。