LeetCode第[15]題(Java)：3Sum （三數之和為目標值）——Medium

本文轉載自查看原文 2018-01-09 00:17 1813 java/ 刷題筆記/ 數據結構與算法/ LeetCode

題目難度：Medium

題目：

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.

翻譯：

給定一個n個整數的數組S，S中是否存在有a b c三個元素使得a+b+c=0?在S中找到所有的非重復組合，使它們的和為0

注意:答案集不能包含重復的數組。例如：[1,5,-6]與[-6,5,1]等數組，答案只出現其中一個

思路：三。。。個。。for循環？？

(⊙x⊙;)。。。就要寫！

Code：311 / 313 test cases passed. —— Time Limit Exceeded 　　　　　時間復雜度:O(N³)

 1     public static List<List<Integer>> threeSum1(int[] nums) {
 2         Arrays.sort(nums);
 3         List<List<Integer>> outList = new ArrayList<List<Integer>>();
 4         for (int i = 0; i < nums.length; i++) {
 5             if ((i > 0 && nums[i] == nums[i - 1]))  // 各循環起始點不需要判斷重復
 6                 continue; // 不用i++ 的原因：避免最后的k還要增加邊界判斷，進入下一個循環則會自動邊界判斷“i < nums.length”
 7             for (int j = i + 1; j < nums.length; j++) {
 8                 if ((j > i + 1 && nums[j] == nums[j - 1])) 
 9                     continue; // 當有很多for和if的時候，條件取反后用continue，以此取代if的{}縮進，使代碼可讀性增加
10                 for (int k = j + 1; k < nums.length; k++) {
11                     if ((k > j + 1 && nums[k] == nums[k - 1]))
12                         continue;
13                     if ((nums[i] + nums[j] + nums[k] == 0)) {
14                         List<Integer> inList = new ArrayList<Integer>();
15                         inList.add(nums[i]);
16                         inList.add(nums[j]);
17                         inList.add(nums[k]);
18                         outList.add(inList);
19                         break;
20                     }
21                 }
22             }
23         }
24         return outList;
25     }

(因為只有在尾巴進行添加並無其他功能，所以采用ArrayList比較實惠和效率)

312測試用例超時：

[82597,-9243,…………此處省略N千個數字]

不想說話，讓蠢人靜靜。。。。

————————————————————————智商分割線————————————————————————

參考答案：74ms——beats85.12%　　　　時間復雜度O(N²)

    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        Arrays.sort(nums);
        
        for (int i = 0; i < nums.length-2; i++) {
            int left = i + 1;
            int right = nums.length - 1;
            if (i > 0 && nums[i] == nums[i - 1])
                continue; // 去掉重復的起點
            while (left < right) {
                int sum = nums[left] + nums[right] + nums[i];
                if (sum == 0) {
                    result.add(Arrays.asList(new Integer[]{nums[i], nums[left], nums[right]}));
                    while (left < right && nums[left] == nums[left + 1])
                        left++; // 去掉重復的左點
                    while (left < right && nums[right] == nums[right - 1])
                        right--; // 去掉重復的右點
                    right--; // 進入下一組左右點判斷
                    left++;
                } else if (sum > 0) {
                    right--; // sum>0 ,說明和過大了，需要變小，所以移動右邊指針
                } else {
                    left++; // 同理，需要變大，移動左指針
                }
            }
        }
        return result;
    }