leetcode 15三数之和 神仙做法?等再刷刷回头看

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        if (nums.length < 3) return Collections.emptyList(); 
        List<List<Integer>> res = new ArrayList<>(); 
        int minValue = Integer.MAX_VALUE; 
        int maxValue = Integer.MIN_VALUE; 
        int negSize = 0, posSize = 0; 
        int zeroSize = 0; 
        for (int v : nums) { 
            if (v < minValue) minValue = v; 
            if (v > maxValue) maxValue = v; 
            if (v > 0) posSize++; 
            else if (v < 0) negSize++; 
            else zeroSize++; 
        }
        if (zeroSize >= 3) res.add(Arrays.asList(0, 0, 0));//输出 三个 0 的情况 
        if (negSize == 0 || posSize == 0) return res;
        //此时minValue一定为负数,maxValue一定为正数
        //如果maxValue > -2*minValue,那么大于 -2*minValue的元素肯定不会是答案,可以排除掉,所以更新maxValue
        if (minValue * 2 + maxValue > 0) maxValue = -minValue * 2; 
        //同理更新minValue
        else if (maxValue * 2 + minValue < 0) minValue = -maxValue * 2;
        //自己构建一个hashmap,值表示元素重复次数,注意java数组默认值为 0
        int[] map = new int[maxValue - minValue + 1];
        int[] negs = new int[negSize];
        int[] poses = new int[posSize];
        negSize = 0;
        posSize = 0;
        for (int v : nums) {
            if (v >= minValue && v <= maxValue) {//只保留在[minValue,maxValue]区间内的元素
                if (map[v - minValue]++ == 0) {//计数加去重
                    if (v > 0) poses[posSize++] = v;//poses数组存所有去重后的正值
                    else if (v < 0) negs[negSize++] = v;//negs数组存所有去重后的负值 
                }
            }
        }
        //正负数两数组排序
        Arrays.sort(poses, 0, posSize);
        Arrays.sort(negs, 0, negSize);
        int basej = 0;
        for (int i = negSize - 1; i >= 0; i--) {//负数数组从后往前遍历
            int nv = negs[i];//nv为当前负数值
            //minp = -nv/2,相当于三元组中另外两元素的平均值,即为另两个元素中较小值的上界,较大值的下界
            int minp = (-nv) >>> 1;
            //定位到正数数组中值刚好小于平均值的元素(这个地方应该还可以优化为使用二分查找)
            while (basej < posSize && poses[basej] < minp) basej++;
            for (int j = basej; j < posSize; j++) {//正数数组从大于等于平均值的元素开始遍历
                int pv = poses[j];//pv 为当前正数值
                int cv = 0 - nv - pv;//cv 为要寻找的另一个值
                //目标值 cv 应该在 [nv,pv] 当中
                //如果不限制cv<=pv,当nv为奇数时,有可能会重复输出
                if (cv >= nv && cv <= pv) {
                    if (cv == nv) {
                        if (map[nv - minValue] > 1)//两个相同的负数和一个正数的情况
                            res.add(Arrays.asList(nv, nv, pv));
                    } else if (cv == pv) {
                        if (map[pv - minValue] > 1)//两个相同的正数和一个负数的情况
                            res.add(Arrays.asList(nv, pv, pv));
                    } else {
                        if (map[cv - minValue] > 0)//三个不同元素的情况
                            res.add(Arrays.asList(nv, cv, pv));
                    }
                } else if (cv < nv) break;//如果 cv 小于 nv了,表明这种情况会在后面寻找,为避免重复输出,跳出循环
            }
        }
        return res;
    }
}

 

posted @ 2019-11-04 14:30  doyi  阅读(183)  评论(0编辑  收藏  举报