LeetCode题解-15-3Sum

解题思路

题目要求找到数组中所有a+b+c=0的数字。朴素算法是三层遍历，复杂度是O(n^3)。
我的思路是首先对数组排序,需要O(nlogn),然后对a,b进行二层遍历并对c进行二分查找，需要O(logn*n^2)。
最终复杂度是O(logn*n^2)，应该还有优化空间。

参考源码

public class Solution {
    public List<List<Integer>> threeSum(int[] num) {
        List<List<Integer>> sums = new ArrayList<List<Integer>>();
        if (num == null || num.length < 3) {
            return sums;
        }

        Arrays.sort(num);
        for (int i = 0; i < num.length - 2 && num[i] <= 0; i++) {
            if (i > 0 && num[i] == num[i - 1]) {
                continue;
            }

            int a = num[i];
            for (int j = i + 1; j < num.length - 1; j++) {
                if (j > i + 1 && num[j] == num[j - 1]) {
                    continue;
                }

                int b = num[j];
                int c = -(a + b);
                if (c < b) {
                    break;
                }

                int index = Arrays.binarySearch(num, j + 1, num.length, c);
                if (index >= 0) {
                    sums.add(Arrays.asList(a, b, c));
                }
            }
        }

        return sums;
    }
}