OJ练习48——T15 3Sum

找一列整数中和为0的三个数,要求返回结果按序排列,且返回的序列之间互不相同。例如:算法

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

【思路】函数

(看来的,本身真是没有一点算法头脑)spa

对于每个当前数,用2sum算法求剩下的两个数和为当前数相反数。3sum退化成2sum问题。指针

因为题目不要求返回原序列的序号,因此能够用2sum的先排序再双指针向中遍历的算法。调试

【other code】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;
    } 

【总结】blog

由于序列中有重复的数字因此要去除重复,j和k是双指针向中搜索。排序

可能会奇怪,若是忽略了重复的数,(-1,-1,2)的状况怎么办。three

因而我分步调试了一下= =

i指向-1时,j=i+1,指向-1,这样就不会漏掉三个数中两个数相同的状况了。O(∩_∩)O~~

另外,stl的sort函数应该是快排吧,时间是O(nlogn)。有库函数很方便,还觉得要本身手动写了……懒癌。

【后记】

虽然耗费了一上午的时间,但总结了T1的2sum,看了编程之美2.12节相关解释,仍是有点收获的。

相关文章
相关标签/搜索