LeetCode 腾讯精选练习50 题——15. 三数之和

题目

解答

直接用暴力遍历搜解的话时间复杂度为O(n^3)，这个代码就不写了。我们这里考虑一下时间复杂度为O(n^2)的解法：

首先对数组nums进行升序排序，然后遍历一次数组（重复的跳过），对于遍历到的每个数字curr找到其右边的数组中是否存在两个数字nums[i]、nums[j]（i < j）使得nums[curr]、nums[i]、nums[j]三者之和为0，这里可以参考我之前写过的两数之和 II - 输入有序数组中的算法，复杂度为O(n)。

需要注意的是，curr右边的数组中可能不止有一组(i，j)使得nums[curr]、nums[i]、nums[j]三者之和为0，所以在上面的两数之和 II - 输入有序数组算法中，我们找到第一组 (i, j) 后，还需要继续遍历curr右边的数组，但是这个时候左右指针i、j的值应该为：

i = i 的右边第一个和num[i]不相等的数字对应的下标；

j = j 的左边第一个和num[j]不相等的数字对应的下标；

代码如下：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        if(nums.size() < 3)
            return ans;
        if(nums.size() == 3) {
            if(nums[0] + nums[1] + nums[2] != 0)
                return ans;
            ans.push_back(nums);
            return ans;
        }
        sort(nums.begin(), nums.end());
        for(int curr = 0; curr < nums.size(); curr++) {
            if(nums[curr] > 0)
                break;
            if(curr > 0 && nums[curr-1] == nums[curr])
                continue;
            int i = curr+1, j = nums.size()-1;
            while(i < j) {
                int temp = nums[i] + nums[j];
                if(temp == -nums[curr]) {
                    vector<int> triple = {nums[curr], nums[i], nums[j]};
                    ans.push_back(triple);
                    while(i < j && nums[i] == nums[i+1])
                        i++;
                    while(i < j && nums[j] == nums[j-1])
                        j--;
                    i++;
                    j--;
                }
                else if(temp > -nums[curr])
                    j--;
                else
                    i++;
            }
        }
        return ans;
    }
};