【icyle】Leetcode-cn：15. 三数之和

题目

解答：排序+双指针

思路（大白话）

假设选定的三个数为a、b、c，那么对于这一个数组[-2,-1,0,1,2]（默认已经排好序），有

-2 -1 0 1 2
↑   ↑     ↑
a   b     c

-2 + (-1) + 2 = -1 ，小了，把b调大一点吧。（因为c已经是最大了，而a是固定的）

-2 -1 0 1 2
↑     ↑   ↑
a     b   c

-2 + 0 + 2 = 0，这回对了。
注：如果结果又大了，把c调小一点即可。当a遍历完之后，自然就结束了。这里还可以加上一个剪枝：如果a大于0，那么之后的a+b+c无论如何都不可能等于0，可以直接退出。

需要的头文件

vector、algorithm

代码

/*
 * @lc app=leetcode.cn id=15 lang=cpp
 *
 * [15] 三数之和
 */
// @lc code=start
#include <vector>
#include <algorithm>
using namespace std;
class Solution
{
public:
    vector<vector<int>> threeSum(vector<int> &nums)
    {
        vector<vector<int>> resnums;
        int numslength = nums.size();
        //特殊情况
        if (numslength < 3)
        {
            return resnums;
        }
        sort(nums.begin(), nums.end());
        //这里排序加双指针法比较好理解，重点是要如何减少查询次数
        //1. 一重循环和二重循环重复的数不再查
        //一重循环比如[1,1,1,3,4,5]，第一个1和第二个1如果都查，那么类似[1,3,4][1,3,5]这种就重复了
        //二重循环比如上面例子的第一个1固定，第一次查找是[1,1]+[1,3,4,5]中的一个，第二次查找是[1,1]+[3,4,5]中的一个，就有重复了
        for (int i = 0; i < numslength - 2; i++)
        {
            if (nums[i] > 0)
                break;
            //注意这里不要漏i>0，因为一开始i=0的话，i-1是非法的
            if (i > 0 && nums[i] == nums[i - 1])
            {
                continue;
            }
            //由于已知target为0，就可以三数之和转化为两数之和=target
            int target = -nums[i];
            int left = i + 1, right = numslength - 1;
            while (left < right)
            {
                if (nums[left] + nums[right] == target)
                {
                    resnums.push_back({nums[i], nums[left], nums[right]});

                    // 这个if循环是找到答案了，那么现在left和right都不需要了，直接移动到下一顺位
                    // 如果出现重复情况，需要再次顺位
                    left++;
                    right--;
                    while (left < right && nums[left] == nums[left - 1])
                        left++;
                    while (left < right && nums[right] == nums[right + 1])
                        right--;
                }
                else if (nums[left] + nums[right] < target)
                {
                    left++;
                }
                else if (nums[left] + nums[right] > target)
                {
                    right--;
                }
            }
        }
        return resnums;
    }
};
// @lc code=end

时间复杂度和空间复杂度

• 时间复杂度： O(n²) ，其中 n 是数组 nums 的长度。虽然代码中只有一重for循环，但对于双指针（while (left < right)）来说，left和right走过的平均距离在最坏情况下也是走了一遍数组的长度。这里给出更官方的解释：
  在枚举的过程每一步中，「左指针」会向右移动一个位置（也就是题目中的 b），而「右指针」（c）会向左移动若干个位置，这个与数组的元素有关，但我们知道它一共会移动的位置数为 O(n) ，均摊下来，每次也向左移动一个位置，因此时间复杂度为 O(n) 。

• 空间复杂度：如果本题不允许修改nums，那么我们需要复制一份相同的数组，然后对其进行排序。复制一份的时间复杂度为 O(n) ，排序的时间复杂度为 O(logn) ，渐进情况下得到总的时间复杂度为 O(n) 。

反思与总结

想一想为什么c指针一定要从末尾开始？