OJ練習48——T15 3Sum

找一列整數中和爲0的三個數，要求返回結果按序排列，且返回的序列之間互不相同。例如：

array S = {-1 0 1 2 -1 -4},
A solution set is:
(-1, 0, 1)    (-1, -1, 2)

【思路】

（看來的，本身真是沒有一點算法頭腦）

對於每個當前數，用2sum算法求剩下的兩個數和爲當前數相反數。3sum退化成2sum問題。

因爲題目不要求返回原序列的序號，因此能夠用2sum的先排序再雙指針向中遍歷的算法。

【other code】

vector<vector<int>> threeSum(vector<int>& num) {
        vector<vector<int> > ret;
        ret.clear();
        sort(num.begin(),num.end());//soga!!
        for(int i=0; i!=num.size();i++){
            if(i > 0 && num[i]==num[i-1])
                continue;//duplicate triplets
            int j,k;
            j=i+1;
            k=num.size()-1;
            while(j<k){
                if(j>i+1&&num[j]==num[j-1]){ 
                    j++;
                    continue;
                }
                if(k<num.size()-1&& num[k]==num[k+1]){
                    k--;
                    continue;
                }
                 int sum = num[i] + num[j] + num[k];
                if(sum>0){
                    k--;
                }else if(sum<0){
                    j++;
                }else{
                    vector<int> tmp;
                    tmp.push_back(num[i]);
                    tmp.push_back(num[j]);
                    tmp.push_back(num[k]);
                    ret.push_back(tmp);
                    j++;
                }
            }
        }
        return ret;
    } 

【總結】

由於序列中有重複的數字因此要去除重複，j和k是雙指針向中搜索。

可能會奇怪，若是忽略了重複的數，(-1,-1,2)的狀況怎麼辦。

因而我分步調試了一下= =

i指向-1時，j=i+1,指向-1，這樣就不會漏掉三個數中兩個數相同的狀況了。O(∩_∩)O~~

另外，stl的sort函數應該是快排吧，時間是O(nlogn)。有庫函數很方便，還覺得要本身手動寫了……懶癌。

【後記】

雖然耗費了一上午的時間，但總結了T1的2sum，看了編程之美2.12節相關解釋，仍是有點收穫的。