LeetCode 15. 3Sum [Medium] [C++] 解題筆記

這題給定一個 array，要我們在其中找出三個數字且其和為 0 的所有組合。

EX:

        nums = [-1, 0, 1, 2, -1, -4]

          [

Ans:     [-1, 0, 1],
            [-1, -1, 2]
          ] 

想法:

    這題是 1. Two Sum 的進階題，只要找兩個數字和時我們可以很簡單的固定一個數字，然後搭配 hashmap 儲存

遍歷過得數字來快速查找是否有解，而這題也一樣，我們可以先固定三數中的一數，然後利用 Two Pointer 的技巧，從

兩端往中間遍歷，若和過大則右邊往左移動，若和過小則左邊往右移動，因此我們一開始必須先做 sorting，而因為題目

要求不要輸出重複的解，因此 sorting 過後我們也可以很容易避免列舉重複的解，每個數字只有在第一次出現時才能用來列舉，

如果前一個數字跟當前數字一樣，則當前數字就不需要再列舉可以直接略過。另外這題另外有個小技巧，因為我們目標是組合出 0，

而我們的 array 在一開始就已經排序好了，所以一旦遍歷到當前數字大於 0 就可以結束了，因為排序過了後面的數字也都大於 0，

不可能再找到和為 0 的解了，算是一種剪枝。

Complexity: O(nlogn + n^2) time

完整程式碼:

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        if (nums.size() < 3) { return {};}
        vector<vector<int>> res;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size() - 2; i++) {
            if (nums[i] > 0) { break; }
            if (i > 0 && nums[i] == nums[i - 1]) { continue; }
            int target = 0 - nums[i];
            int left = i + 1;
            int right = nums.size() - 1;
            while (left < right) {
                int sum = nums[left] + nums[right];
                if (target == sum) {
                    res.push_back({nums[i], nums[left], nums[right]});
                    while (left < right && nums[left] == nums[left + 1]) { left++; }
                    while (left < right && nums[right] == nums[right - 1]) {
                        right--;
                    }
                    left++;
                    right--;
                }
                else if (target > sum) {
                    left++;
                }
                else {
                    right--;
                }
            }
        }
        return res;   
    }
};