LeetCode第15题——三数之和（medium）

题目链接

三数之和

解法：排序+双指针（$O(n^2)$）

为了去重和双指针遍历方便，我们先将数组进行排序操作。由于是“三数之和”，本题事实上需要三个指针指向三个位置的数字。但三个指针不太方便我们遍历，所以我们采用的做法是遍历第一个指针$i$，对于每一个确定的$i$，用双指针操作$j$和$k$​来寻找答案。

OK，现在我们在一个已经排好序的升序数组中叙述我们的做法。

$i$的操作

$i$要做的事非常简单，就是作为最外层的循环从小到大遍历数组元素。但是注意，我们不能找重复的三元组，也就是说对于相同值的元素，$i$需要且只需要指向它们任意一个，让后面的双指针跑一次就够了。由于这是升序数组，我们具体做法是如果当前位置不为0，且与上一个元素的值相等，则$i$跳过它去指向下一个。

$j$和$k$的操作

第一个确定的指针$i$像是一条界限，把后面的指针$j$和$k$的范围限定在[i + 1, n]上。初始时，我们让$j$指向$i+1$的位置上，$k$指向末尾$n$的位置，接着开始经典操作：两个指针向中间遍历。

• 对于$j$而言，它和$i$一样需要去重，做法同样是如果当前位置不是初始的$i+1$，且与上一个元素的值相等，则$j$跳过它去指向下一个。这样，每次我们就得到了一个唯一且确定的指针$j$。
• 对于$k$而言，由于数组是升序排列的，所以它越往左走，三数之和就越小。所以，当$k=k-1$时三数之和如果为非负，就让$k$一直向左走下去。这个操作一来是找答案，也自带对$k$的去重效果。
• 如果我们最后找到一个符合要求的三元组{i, j, k}，接下来的操作是$i$或$j$向后走。如果是$j$向后走直到下一个确定的值，三数之和必然大于0。所以这时候，我们没必要让$k$从末尾开始走，接着上次的位置继续向左遍历就行。
• 当然，上述一切操作的前提都是j < k。

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int> > res;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size(); i ++) {
            if (i && nums[i] == nums[i - 1]) continue;
            for (int j = i + 1, k = nums.size() - 1; j < k; j ++) {
                if (j > i + 1 && nums[j] == nums[j - 1]) continue;
                while (j < k - 1 && nums[i] + nums[j] + nums[k - 1] >= 0) k --;
                if (nums[i] + nums[j] + nums[k] == 0)
                    res.push_back({nums[i], nums[j], nums[k]});
            }
        }
        return res;
    }
};