LeetCode题解——3Sum

题目：

给定一个数组，找出其中和为0的所有3个数的组合。每个组合的3个数都是非递降的。

 

解法：

先排序再遍历，设置3个指针，第一个依次遍历，第二三个在第一个指针后面的部分里，左右夹逼查找和为第一个数的相反数的组合。时间O(N²)。

 

代码：

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        vector<vector<int> > result;
        if(num.size() < 3)
            return result;
            
        sort(num.begin(), num.end());　　//排序
        
        for(int a = 0; a < num.size() - 2; ++a)　　//a指向第一个数
        {
            if(num[a] > 0)　　//如果第一个数>0
                break;
                
            if(a > 0 && num[a] == num[a-1])　　//如果和上一个重复，则前进一个
                continue;
                
            for(int b = a + 1, c = num.size() - 1; b < c; ) //b、c分别指向第二个数和第三个数，左右夹逼
            {
                if(b > a + 1 && num[b] == num[b-1])　　//和上一个重复，则前进1
                {
                    ++b;
                    continue;
                }
                if(c < num.size() - 1 && num[c] == num[c+1])　　//同上
                {
                    --c;
                    continue;
                }
                
                int sum = num[a] + num[b] + num[c];
                if(sum == 0)
                {
                    result.push_back({num[a], num[b], num[c]});
                    ++b; 
                    --c;
                }
                else if(sum > 0)39                     --c;
                else
                    ++b;
            }
        }
        
        return result;
    }
};