【leetcode系列】3Sum

標籤：leetcode   3sum   

這個題我最開始的思路是：先一個數定下來，然後在除這個數之外的集合裡面找另外兩個數，最後計算和。如此反覆，對於N個數，需要進行N-2次迴圈。

我遇到的問題就是怎麼找另外兩個數，其實我想過參照Two Sum裡面的解法，就是用Hashtable存，鍵值對的結構是<任意兩個數的和，<下標1，下標2>>，但是構造這個Hashtable就需要O(N^2)，後面真正解的時候有需要O(N^2)。

參考了大牛的解法後，明白了找兩個數還是用兩個下標同時往中間移動比較好，下面上代碼。

import java.util.ArrayList;import java.util.Arrays;public class Solution {public static ArrayList<ArrayList<Integer>> threeSum(int[] numbers) {ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();if (numbers.length < 3) {return result;}Arrays.sort(numbers);for (int i : numbers) {System.out.print(i + " ");}System.out.println();for (int i = 0; i < numbers.length - 2; i++) {int lp = i + 1;int rp = numbers.length - 1;while (lp < rp) {int sum = numbers[i] + numbers[lp] + numbers[rp];if (sum == 0) {ArrayList<Integer> tmp = new ArrayList<Integer>();tmp.add(numbers[i]);tmp.add(numbers[lp]);tmp.add(numbers[rp]);result.add(tmp);do {lp++;} while (lp < rp && numbers[lp] == numbers[lp + 1]);do {rp--;} while (lp < rp && numbers[rp] == numbers[rp - 1]);} else if (sum < 0) {lp++;} else if (sum > 0) {rp--;}}}return result;}public static void main(String[] args) {int[] a = { -1, 0, 1, 2, -1, -4 };ArrayList<ArrayList<Integer>> result = threeSum(a);for (ArrayList<Integer> item : result) {for (Integer i : item) {System.out.print(i + " ");}System.out.println();}}}