[LeetCode] 3Sum

这个问题还是《编程之美》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;
    }
}