LeetCode 15. 三数之和

思路

方法：排序 + 双指针

先排序，之后有序数组，固定一个数，对后面的有序序列进行双指针查找，双指针方法见此题：LeetCode 167. 两数之和 II - 输入有序数组

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        if(nums.size() < 3) {
            return res;
        }

        //从小到大排序
        sort(nums.begin(), nums.end());
        int n = nums.size();
        for(int i = 0; i < n; ++i) {
            if(nums[i] > 0) {
                //因为是从小到大排好了序，所以遇到的数大于0，后面全都会大于0，肯定不会出现三数之和等于0的情况了
                break;  
            }
            
            if(i > 0 && nums[i] == nums[i-1]) 
                continue;   //去重

            int L = i+1, R = n-1;
            while(L < R) {
                int sum = nums[i] + nums[L] + nums[R];
                if(sum == 0) {
                    vector<int> v({nums[i], nums[L], nums[R]});
                    res.push_back(v);
                    while(L < R && nums[L] == nums[L+1]) L++; //去重
                    while(L > R && nums[R] == nums[R-1]) R--;   //去重
                    L++;
                    R--;
                } else if (sum < 0) {
                    L++;
                } else {
                    R--;
                }
            }

        }

        return res;
    }
};

复杂度分析

时间复杂度：O(n²)，排序是O(nlogn)，固定一个数，对后序进行双指针查找，这个过程是O(n²)，所以总体是O(n²)。

原文：画解算法：15. 三数之和