leetcode解题方案--015--3 sum

最新推荐文章于 2024-03-28 20:32:43 发布

笔筒谜语

最新推荐文章于 2024-03-28 20:32:43 发布

阅读量228

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/sinat_16014143/article/details/78285396

版权

leetcode 专栏收录该内容

85 篇文章 0 订阅

订阅专栏

题目

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.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

分析

题目隐含的条件，不能有重复的list，不能每个小list必须从小到大排序。

我的思路有两个
1 将n^3转化为2sum。leetcode第一题就是一道2sum，我用hashmap做的
2 使用数组左右指针。

实践来看，第二种方法更优。

这个有两个测试例没过。
稍后我会分析一下时间长的原因。

class Solution {
     public List<List<Integer>> threeSum(int[] nums) {

        HashSet<List<Integer>> set = new HashSet<>();
        List<List<Integer>> list = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
            Map<Integer, Integer> map = new HashMap<Integer, Integer>();
            for (int k = i + 1; k < nums.length; k++) {
                if (map.containsKey(nums[k])) {

                    List<Integer> tmpList = new ArrayList<>();
                    tmpList.add(nums[i]);
                    tmpList.add(nums[k]);
                    tmpList.add( 0 - nums[i] - nums[k]);
                    Collections.sort(tmpList);
                    set.add(tmpList);
                } else {
                    map.put(0 - nums[i] - nums[k], k);
                }
            }
        }
        list.addAll(set);
        return list;
    }
}

这是左右指针的方法，去掉了放进set去重的部分，时间更短。
那如何保证不重复呢，关键点在于排序。一个有序的数组，相同的数就会相邻。
在这个程序中 if (i == 0 || (i > 0 && nums[i] != nums[i-1]))会去掉-sum相同的情况，而while (k1 < k2 && nums[k1] == nums[k1+1]) k1++;
while (k1 < k2 && nums[k2] == nums[k2-1]) k2–;会去掉指针移动时相同的情况。

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

        Arrays.sort(nums);
        List<List<Integer>> list = new ArrayList<>();

        for (int i = 0; i < nums.length - 2; i++) {
            if (i == 0 ||  (i > 0 && nums[i] != nums[i-1])) {
                int k1 = i + 1;
                int k2 = nums.length - 1;
                while (k2-k1>=1) {
                    if (nums[i] + nums[k1] + nums[k2] == 0) {
                        list.add(Arrays.asList(nums[i], nums[k1], nums[k2]));
                        while (k1 < k2 && nums[k1] == nums[k1+1]) k1++;
                        while (k1 < k2 && nums[k2] == nums[k2-1]) k2--;
                        k1++;k2--;
                    } else if (nums[i] + nums[k1] + nums[k2] > 0) {
                        k2--;
                    }else {
                        k1++;
                    }
                }
            }
        }
        return list;
    }