【LeetCode】3Sum 解决报告

这个问题是我目前的知识回答，不来，只有良好的网上搜索解决方案，发现 K Sum 它是一类问题，但是，互联网是没有更简洁的代码，我想对于谁刚开始学习的人。您可能仍然想看看这个问题该怎么解决，然后看看他们的利益扩展它。

【称号】

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:

Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
The solution set must not contain duplicate triplets.

    For example, given array S = {-1 0 1 2 -1 -4},

    A solution set is:

    (-1, 0, 1)

    (-1, -1, 2)

【解析】

K Sum问题是一个系列，博客 http://tech-wonderland.net/blog/summary-of-ksum-problems.html 总结得比較完整，有兴趣能够去看。

作为刚開始学习的人。我们用最简洁的思路来说一下这道题怎么解。

暴力解决法是每一个人都能想到的。三层for循环，时间复杂度是O(n^3)，并且还要处理反复的问题，显然不是题目想要的解法。

那能不能降到O(n^2)？排序算法的时间复杂度为O(nlgn)，小于O(n^2)，那么我们最好还是先对数组排个序。

排序之后，我们就能够对数组用两个指针分别从前后两端向中间扫描了，假设是 2Sum，我们找到两个指针之和为target就OK了。那 3Sum 类似。我们能够先固定一个数，然后找另外两个数之和为第一个数的相反数就能够了。

代码不难。先看了再说。

【Java代码】O(n^2)

public class Solution {

    List<List<Integer>> ret = new ArrayList<List<Integer>>();

    public List<List<Integer>> threeSum(int[] num) {

        if (num == null || num.length < 3) return ret;

        Arrays.sort(num);

        int len = num.length;

        for (int i = 0; i < len-2; i++) {

            if (i > 0 && num[i] == num[i-1]) continue;

            find(num, i+1, len-1, num[i]); //寻找两个数与num[i]的和为0

        }

        return ret;

    }

    public void find(int[] num, int begin, int end, int target) {

        int l = begin, r = end;

        while (l < r) {

            if (num[l] + num[r] + target == 0) {

                List<Integer> ans = new ArrayList<Integer>();

                ans.add(target);

                ans.add(num[l]);

                ans.add(num[r]);

                ret.add(ans); //放入结果集中

                while (l < r && num[l] == num[l+1]) l++;

                while (l < r && num[r] == num[r-1]) r--;

                l++;

                r--;

            } else if (num[l] + num[r] + target < 0) {

                l++;

            } else {

                r--;

            }

        }

    }

}

注意，对于 num[i]，寻找另外两个数时，仅仅要从 i+1 開始找就能够了。

这样的写法，能够避免结果集中有反复，由于数组时排好序的。所以当一个数被放到结果集随着时间的推移，它的背后，其相当于直接跳过。

秒客网

【LeetCode】3Sum 解决报告

相关文章