LeetCode-15 3Sum笔记

题目描述

给定一个包含n个整数的数组nums，在nums中是否存在三个元素a，b，c满足a+b+c = 0 ?找到数组中所有满足和为0这一条件的三元组，三元组之间不能有重复。

示例：

Given array nums = [-1, 0, 1, 2, -1, -4],
A solution set is:
[
  [-1, 0, 1],  [-1, -1, 2]
]

解题思路

1. 第一感觉

看到这道题的第一感觉这是一个组合问题，即从n个数中选3三个数。将所有可能的三元组求出来并判断是否满足和为0的条件即可解决问题。代码如下：用一个三重循环来求所有可能的组合

for (int i = 0; i < nums.size() - 2; ++i)
    for (int j = i + 1; j < nums.size() - 1; ++j)
        for (int k = j + 1; k < nums.size(); ++k)
            if (nums[i] + nums[j] + nums[k] == 0)
            {
                vector<int> ret;
                ret.push_back(nums[i]);
                ret.push_back(nums[j]);
                ret.push_back(nums[k]);    
                rets.push_back(ret);
            }

但注意到数组中的元素可能有重复，因此所有可能的三元组中会有一些重复。为了避免重复，我们可以首先对数组排序。这样在遇到重复的元素后，可以直接跳过，代码如下：

sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size() - 2; ++i)
{		
    if (i > 0 && nums[i] == nums[i - 1])
        continue;
    for (int j = i + 1; j < nums.size() - 1; ++j)
    {
        if (j > i + 1 && nums[j] == nums[j - 1])
            continue;
        for (int k = j + 1; k < nums.size(); ++k)
        {
            if (k > j + 1 && nums[k] == nums[k - 1])
            continue;
            if (nums[i] + nums[j] + nums[k] == 0)
            {
                vector<int> ret;
                ret.push_back(nums[i]); 
                ret.push_back(nums[j]); 
                ret.push_back(nums[k]);
                rets.push_back(ret);
            }
        }
    }
}

可以提交运行一下，确超时了，看来时间复杂度O(n3)是不行的。

2. 网上求助

O(n2)的解法（参考3Sum的维基百科词条）。

首先，我们还是先对数组排序。接下来，

（1）和刚才O(n3)的解法一样，先确定三元组（a，b，c）中a的位置；

（2）在a右侧的元素中，找到两个b，c满足b + c == -a。那么如何来查找b，c就是O(n2)的解法和O(n3)的解法的区别了；

b，c可以用两个指针相向移动的方法来查找，首先给出代码

int start = i + 1;
int end = nums.size() - 1;
while (start < end)
{
    int b = nums[start];
    int c = nums[end];
    if (a + b + c == 0)
    {
        vector<int> ret;
        ret.push_back(a); ret.push_back(b); ret.push_back(c);
        rets.push_back(ret);
        end--;
        while (end > start && nums[end] == nums[end+1] )
        end--;
    }
    else if (a + b + c > 0)
    {
        end--;
        while (end > start && nums[end] == nums[end + 1])
        end--;
    }
    else
    {
        start++;
        while (end > start && nums[start] == nums[start - 1])
        start++;
    }
}

代码很简单，但是关键在于为什么让start和end相向而行，就一定能找到所有满足条件的b，c呢？对此，可以分情况讨论

当a + b + c == 0时，这时候为了找到下一个满足条件的b，c，需要改变start和end的位置，这时可以选择start或end中的任意一个。因为只要start和end中某一个离开了原位置，b + c就不可能等于-a；

当a + b + c > 0时，这时候为了找到满足条件的b，c，需要让b+c的值变得小一点，因此，可以将end向左移动（因为数组已经拍排好序了，start不动，end向左移动，b+c的值肯定会变小）；

当a + b + c < 0时，同理我们可以将start向右移动。

通过以上分析，可以看出，如果存在a + b + c == 0，上述方法一定会向着a+b+c == 0的方向进行。而这样来寻找b，c的过程时间复杂度只有O(n)，因此加上外面确定a的那层循环，算法整体复杂度为O(n2)。

下面是完整代码：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> rets;
        if (nums.size() < 3)
            return rets;
        sort(nums.begin(), nums.end());

        for (int i = 0; i < nums.size() - 2; ++i)
        {
            int a = nums[i];
            if (i > 0 && a == nums[i - 1])
                continue;
            int start = i + 1;
            int end = nums.size() - 1;
            while (start < end)
            {
                int b = nums[start];
                int c = nums[end];
                if (a + b + c == 0)
                {
                    vector<int> ret;
                    ret.push_back(a); ret.push_back(b); ret.push_back(c);
                    rets.push_back(ret);
                    end--;
                    while (end > start && nums[end] == nums[end+1] )
                        end--;
                }
                else if (a + b + c > 0)
                {
                    end--;
                    while (end > start && nums[end] == nums[end + 1])
                        end--;
                }
                else
                {
                    start++;
                    while (end > start && nums[start] == nums[start - 1])
                        start++;
                }
            }
        }
        return rets;
    }
};