LetCode 15. 三数之和

static const auto ______ = [](){ 
    std::cout.sync_with_stdio(false);
    cin.tie(0);
    return 0;
}();

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        if (nums.size() < 3)
            return vector<vector<int>>();
        // 先排序
        sort(nums.begin(), nums.end());
        vector<vector<int>> ans;
        int a, b, c;
        int size = nums.size();
        for (a = 0; a < size - 2; ++a){
            // 如果a大于0了，直接返回
            if (nums[a] > 0)
                return ans;
            b = a + 1;
            c = size - 1;
            while(b < c){
                // 如果符合要求
                if (nums[a] + nums[b] + nums[c] == 0){
                    ans.push_back({nums[a], nums[b], nums[c]});
                    b++,c--;
                    // 跳过和b相同的数字
                    while (b < c && nums[b] == nums[b - 1]) b++;
                    // 跳过和c相同的数字
                    while (b < c && nums[c] == nums[c + 1]) c--;
                }else if (nums[a] + nums[b] + nums[c] < 0 )
                    b++;
                else
                    c--;
            }
            // 跳过和a相同的数字
            while(a < size && nums[a + 1] == nums[a]) a++;
        }
        return ans;
    }
};