LeetCode系列(二)-3Sum

  给定一个包含 n 个整数的数组 nums,
  判断 nums 中是否存在三个元素 a,b,c ,
  使得 a + b + c = 0 ?
  找出所有满足条件且不重复的三元组。

示例:

  给定数组 nums = [-1, 0, 1, 2, -1, -4],
  满足要求的三元组集合为:
  [
   [-1, 0, 1],
   [-1, -1, 2]
  ] 

解法:

1. 时间复杂度O(n^2)

思路:双指针求指定和。优化:去重判断;剪枝优化。

        List<List<Integer>> list = new ArrayList<>();
        Arrays.sort(nums);
        for (int i = 0; i < nums.length-2; i++) {
            // 第一个数去重 如:-4, -1, -1, 0, 1, 2 数组。当第二个数 -1=第三个数 -1。那么第三个数-1得到的情况<=第二个数-1的情况。会有重复。
            if (i==0||(i>0&&!(nums[i]==nums[i-1]))) {
                int target = -nums[i];
                // 剪枝优化,因为排序后。如果第一个数为正数,则第二第三个数也会为正数。就不会有相加等于0的情况。
                if (nums[i] > 0) {
                    return list;
                }
                // 采用双指针法
                int lo = i + 1;
                int hi = nums.length - 1;
                while (lo < hi) {
                    if (nums[lo] + nums[hi] == target) {
                        list.add(Arrays.asList(nums[i], nums[lo], nums[hi]));
                        // 二三数 去重,如果之后的元素和当前元素一样的话,就跳过他。因为得到的结果一样。
                        while (lo < hi && nums[lo] == nums[lo + 1]) lo++;
                        while (lo < hi && nums[hi] == nums[hi - 1]) hi--;
                        lo++;
                        hi--;
                    } else if (nums[lo] + nums[hi] > target) {
                        hi--;
                    } else {
                        lo++;
                    }
                }
            }
        }
        return list;    

 

posted @ 2018-09-11 11:34  天才小彩笔i  阅读(257)  评论(0编辑  收藏  举报