[LeetCode] 3Sum

来源：互联网发布：免费开淘宝店流程视频编辑：程序博客网时间：2024/06/06 02:16

这个问题还是《编程之美》2.12的扩展问题。

最一开始的思路是，第一个数肯定是小于等于0，最后一个数是大于等于0，中间的数不确定，于是想着在这上面做点文章，但是后来遇到些问题，好像不太可行。

然后搞了一个O(N^3)的算法，超时了。于是又用了一个HashMap，把复杂度降到O(N^2)，另外还用了两个标志位set标记重复的可能。用了很多Java的容器，但我对这些容器都不太熟悉，要翻翻书才能做出来。很多排序和重复数据处理的工作都让容器做了，但是这些应该都不影响程序的复杂度。

不太喜欢这类题目，对于我来说是体力活的解法，暂时没有想出特别好的技巧，因为不管怎样都要遍历两遍的吧。

public class Solution {    public ArrayList<ArrayList<Integer>> threeSum(int[] num) {int length = num.length;HashMap<Integer, TreeSet<Integer>> map = new HashMap<Integer, TreeSet<Integer>>();for (int i = 0; i < length; ++i) {if (map.containsKey(num[i])) {map.get(num[i]).add(i);} else {TreeSet<Integer> s = new TreeSet<Integer>();s.add(i);map.put(num[i], s);}}Set<ArrayList<Integer>> temp = new HashSet<ArrayList<Integer>>();Set<Integer> mark1 = new HashSet<Integer>();for (int i = 0; i < length; i++) {int num1 = num[i];if (mark1.contains(num1)) {continue;} else {mark1.add(num1);}Set<Integer> mark2 = new HashSet<Integer>();for (int j = i + 1; j < length; j++) {int num2 = num[j];if (mark2.contains(num2)) {continue;} else {mark2.add(num2);}int num3 = 0 - num1 - num2;if (map.containsKey(num3)) {TreeSet<Integer> x = map.get(num3);mark2.add(num3);if (x.last() > j) {ArrayList<Integer> r = new ArrayList<Integer>();r.add(num1);r.add(num2);r.add(num3);Collections.sort(r);temp.add(r);}}}}ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();result.addAll(temp);return result;    }}

0 0