Leetcode - 3Sum

Question:

Paste_Image.png

My code:

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        if (nums == null)
            return null;
        Arrays.sort(nums);
        
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        if (nums.length < 3 || nums[0] > 0 || nums[0] + nums[1] + nums[2] > 0)
            return result;
        
        for (int i = 0; i < nums.length - 2; i++) {
            int negate = -nums[i];
            int start = i + 1;
            int end = nums.length - 1;
            while ((i == 0 || nums[i] != nums[i - 1]) && start < end) {
                if (nums[start] + nums[end] > negate) {
                    end--;
                }
                else if (nums[start] + nums[end] < negate) {
                    start++;
                }
                else {
                    List<Integer> s = new ArrayList<Integer>();
                    s.add(nums[i]);
                    s.add(nums[start]);
                    s.add(nums[end]);
                    result.add(s);
                    start++;
                    end--;
                    while (start < end && nums[start] == nums[start - 1])
                        start++;
                    while (start < end && nums[end] == nums[end + 1])
                        end--;
                }
            }
        }
        return result;
    }
    
    public static void main(String[] args) {
        Solution test = new Solution();
        int[] a = {-1, 0, 1, 2, -1, -4};
        System.out.println(test.threeSum(a));
    }
}

My test result:

Paste_Image.png

這次作業,還是沒能自己做出來。看了提示,說是用雙指針。我也猜到。
于是第一步做對了,先排個序。
然后我錯誤得將該數組分成兩類, < 0 為一類, >= 0 為一類。
然后設置了兩個指針 head, tail分別指向這個數組的頭和尾,想通過這樣的雙指針來解決問題。
同時,我參考了2Sum的解法,先將head and tail 所指的數的和的相反數求出來,存入哈希表中。然后如果小于0,那就移動head指針來找。如果大于0,那就移動tail指針來找。
但是問題來了,當等于0時,如果找到了(nums[middle] == 0 is true),那么接下來, 可能性就會有三種了,要么head++; 要么 tail--. 要么也可以同時加減。我的一個簡單的循環,類似于一個狀態機,無法在一個時刻,在無先決條件的情況下,描述出這三種發展可能性。所以我的程序是錯誤的,或者說,我的模型,再次錯誤了。同時,我還把那個相反數存入了哈希表。下次新的相反數出現時,如果哈希表中已經存在了,那么就直接跳過。以此來篩選掉那些重復情況。這樣其實也是錯的。
比如 4 + -3 = 1; 哈希表中存放 -1. 3 + -2 = 1; 此時完全是另外一種情況,但1已經存在于哈希表中了,于是直接跳過。造成錯誤。
然后這個時候我就放棄了。上網看了下思路。
發現該用這種解法。我也不知道為什么自己想不出這種解法。一個很大的可能是。昨天的水桶題也是雙指針,雙指針是從頭和尾巴向中間走。于是我就直接把那個模型套過來了。
而這道題目,雖然也是頭指針和尾指針向著中間靠,但準確的說,是有三個指針的。是不一樣的模型。
累死了。

**
總結:所以說,雙指針目前碰到的幾個類型。
1.徹底的頭指針,尾指針。 水桶題。
2.一個基指針,然后跟著頭指針和尾指針,此道題目就是如此。
3.也有頭指針再加中間一個指針,同時開始跑。具體題型忘了。
然后,
List<List<Integer>> result = new List<List<Integer>>(); 是錯的。
List<List<Integer>> result = new ArrayList<List<Integer>>(); 是對的。
因為 List是一個接口,不能被new。所以只能用ArrayList來代替。
只有類才能new 接口是不能new的。

那么,接口和抽象類有什么區別。
抽象類可以實現方法。接口不行。
抽象類可以有 static final, 接口不行。
一個子類只能繼承一個抽象類,但可以實現多個接口。

差不多就總結這些吧。媽的每次凌晨做的題目都好帶勁!可惜精神狀態都不是最佳。
**

Anyway, Good luck, Richardo!

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;

public class Solution {
    /* Solution 1: O(n^k) k = 2, 3, 4, 5, ... general solution
    public List<List<Integer>> threeSum(int[] nums) {
        ArrayList<List<Integer>> ret = new ArrayList<List<Integer>>();
        if (nums == null || nums.length <= 2)
            return ret;
        HashSet<List<Integer>> retSet = new HashSet<List<Integer>>();
        ArrayList<Integer> group = new ArrayList<Integer>();
        helper(nums, 3, 0, 0, retSet, group);
        for (List<Integer> tmp : retSet) {
            ret.add(tmp);
        }
        return ret;
    }
    
    private void helper(int[] nums, int k, int begin, int target, HashSet<List<Integer>> retSet, ArrayList<Integer> group) {
        if (k == 0) {
            int sum = 0;
            int[] arr = new int[group.size()];
            for (int i = 0; i < group.size(); i++) {
                arr[i] = group.get(i);
                sum += arr[i];
            }
            if (sum == target) {
                Arrays.sort(arr);
                ArrayList<Integer> newGroup = new ArrayList<Integer>();
                for (int i = 0; i < arr.length; i++)
                    newGroup.add(arr[i]);
                retSet.add(newGroup);
            }
            return;
        }
        for (int i = begin; i < nums.length; i++) {
            group.add(nums[i]);
            helper(nums, k - 1, i + 1, target, retSet, group);
            group.remove(group.size() - 1);
        }
    }
    */
    
    /* Solution 2: sort to simplify, O(n^k) with a smaller constant
    public List<List<Integer>> threeSum(int[] nums) {
        ArrayList<List<Integer>> ret = new ArrayList<List<Integer>>();
        if (nums == null || nums.length <= 2)
            return ret;
        Arrays.sort(nums);
        HashSet<List<Integer>> retSet = new HashSet<List<Integer>>();
        ArrayList<Integer> group = new ArrayList<Integer>();
        helper(nums, 3, 0, 0, 0, retSet, group);
        for (List<Integer> tmp : retSet) {
            ret.add(tmp);
        }
        return ret;
    }
    
    private void helper(int[] nums, int k, int begin, int target, int sum, HashSet<List<Integer>> retSet, ArrayList<Integer> group) {
        if (k == 0) {
            if (sum == target) {
                retSet.add(new ArrayList<Integer>(group));
            }
            return;
        }
        for (int i = begin; i < nums.length; i++) {
            if (sum + nums[i] > target)
                break;
            group.add(nums[i]);
            helper(nums, k - 1, i + 1, target, sum + nums[i], retSet, group);
            group.remove(group.size() - 1);
        }
    }
    */
    
    /* Solution 3: sort, use 2Sum model, O(n^2) with larger constant because of Hash operation
    public List<List<Integer>> threeSum(int[] nums) {
        ArrayList<List<Integer>> ret = new ArrayList<List<Integer>>();
        if (nums == null || nums.length <= 2)
            return ret;
        Arrays.sort(nums);
        HashSet<List<Integer>> retSet = new HashSet<List<Integer>>();
        for (int i = 0; i < nums.length; i++) {
            helper(nums, i, retSet);
        }
        for (List<Integer> tmp : retSet) {
            ret.add(tmp);
        }
        return ret;
    }
    
    private void helper(int[] nums, int targetIndex, HashSet<List<Integer>> retSet) {
        HashMap<Integer, Integer> tracker = new HashMap<Integer, Integer>();
        int target = (-1) * nums[targetIndex];
        for (int i = targetIndex + 1; i < nums.length; i++) {
            if (tracker.containsKey(nums[i])) {
                ArrayList<Integer> group = new ArrayList<Integer>();
                group.add((-1) * target);
                group.add(nums[tracker.get(nums[i])]);
                group.add(nums[i]);
                retSet.add(group);
            }
            else {
                if (target - nums[i] < nums[i])
                    break;
                tracker.put(target - nums[i], i);
            }
        }
    }
    */
    
    /** Solution 4: pass all online tests. O(n ^ 2) with smaller constant */
    public List<List<Integer>> threeSum(int[] nums) {
        ArrayList<List<Integer>> ret = new ArrayList<List<Integer>>();
        if (nums == null || nums.length <= 2)
            return ret;
        Arrays.sort(nums);
        for (int i = 0; i < nums.length; i++) {
            if (i >=1 && nums[i] == nums[i - 1])
                continue;
            int target = (-1) * nums[i];
            int start = i + 1;
            int end = nums.length - 1;
            while (start < end) {
                int sum = nums[start] + nums[end];
                if (sum < target)
                    start++;
                else if (sum > target)
                    end--;
                else {
                    ArrayList<Integer> group = new ArrayList<Integer>();
                    group.add(nums[i]);
                    group.add(nums[start]);
                    group.add(nums[end]);
                    ret.add(group);
                    start++;
                    end--;
                    while (nums[start] == nums[start - 1] && start < end)
                        start++;
                    while (nums[end] == nums[end + 1] && start < end)
                        end--;
                }
            }
        }
        return ret;
    }
    
    
    
    public static void main(String[] args) {
        Solution test = new Solution();
        int[] nums = new int[] {-1, 0, 1, 2, -1, -4};
        System.out.println(test.threeSum(nums));
    }
}

我自己寫了三種做法,但是沒有一個通過測試。
做法一是一個通解,對 kSum 都可以求解。但是沒有排序,復雜度為 O(n ^ 3) 并且常數系數比較大。
做法二是對做法一的優化。先排序,之后當發現和已經大于了要求時表示再也找不到了,就立刻停止。復雜度仍然為O(n ^ 3),但是常數系數會小一些。
做法三是利用2Sum的模型。先固定住一個值,那么后面就是2Sum,target = -nums[i]
然后復雜度是O(n ^ 2) ,因為2Sum是O(n). 但是涉及到哈希表的許多操作,所以常數系數不會小。

然后我看了能通過測試的做法,即做法4.
http://www.programcreek.com/2012/12/leetcode-3sum/
復雜度也是O(n ^ 2), 但是沒有任何哈希操作,所以常數系數小,可以通過測試。
也是固定住一個值, target = (-1) * nums[i]. 然后從頭尾同時開始搜索這個target,一旦發現,加入list。然后呢,為了避免重復,start和end需要向左和向右滑行一段,保證下一對正確的值不再重復。
同時,在最外層for循環中,為了避免重復,每次需要判斷當前nums[i] 是否和
nums[i - 1] 相等。如果相等,那么之后找出來的解很有可能是重復的。所以要避免這種情況。

感覺這道題木還是很經典的。
我一開始直接用DFS的思路來求解,寫起來是簡單,但是復雜度較高。
后來想用2sum來降低復雜度,但是還是存在大量的哈希操作。
最后,用雙指針,頭尾跑,解決了問題。
雙指針,頭尾跑,經常應用于已經排好序的數組中。
留意。

Anyway, Good luck, Richardo!

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容

  • 背景 一年多以前我在知乎上答了有關LeetCode的問題, 分享了一些自己做題目的經驗。 張土汪:刷leetcod...
    土汪閱讀 12,776評論 0 33
  • Question: My code: My test result: 和昨天的題目差不多,3Sum.只不過這個不是...
    Richardo92閱讀 629評論 1 1
  • My code: 一開始我是拿brute force來求解的。后來突然想起來了可以排序。然后這個做法就自然而然出來...
    Richardo92閱讀 202評論 0 0
  • 2 最近幾天一直覺得不舒服,但是我一直強忍著,自己找點藥吃,每天以樂觀的心態對待身邊的每一個人。早...
    黯黯紅塵一路相伴閱讀 869評論 0 9
  • 從很多年前開始,過年除了看春晚、走親戚之外,又多了一個項目,放火! 去年,意外找到了一條被廢棄的小河,河中竟是枯死...
    Yizuifangxiu閱讀 318評論 0 0