LeetCode:三数之和

最新推荐文章于 2024-05-06 20:29:30 发布

Black_Chocolate.

最新推荐文章于 2024-05-06 20:29:30 发布

阅读量374

点赞数

文章标签： leetcode 算法 c++

本文链接：https://blog.youkuaiyun.com/YuDuna/article/details/130666936

版权

文章介绍了LeetCode第15题的解决方案，通过将三数之和问题转化为两数之和问题，降低时间复杂度至O(N^2)。首先对数组排序，然后遍历负数区间找到数对[a,b]，同时在正数区间寻找c，使得a+b+c=0。为了避免重复，需去除相同的数对，并在遍历时跳过重复的元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

15. 三数之和 - 力扣（Leetcode）

分析：首先最直接的做法就是依次枚举可能的情况，时间复杂度为O（N^3）---不妥

优化：根据式子 a+b+c=0,我们可以变化成这种形式：a+b=-c。这样，我们可以找一组数[a,b]去与c做比较即可。这样可以将时间复杂度优化为O（N^2）

注意的点：我们需要去掉重复的数对

方法：在负数区间找C，找除开负数区间找数对[a,b]比较即可。

找数对的方法，我们可以在数组的负数区间进行找或者在正数区间进行找（这里是找的正数区间）

找区间：我们可以先将整个数组进行一次排序，找到nums[i]>0的位置；标记数对查找区间[i+1,nums.size()-1]

去重：如果下一个位置的数与当前位置的数相同，那么就跳过下一个位置。（正/负区间都是如此操作）

代码：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>>ans;
        for(int i=0;i<nums.size();i++){
            if(nums[i]>0) break;
            int l=i+1,r=nums.size()-1;
            while(l<r){
                if(nums[l]+nums[r]==-nums[i]){
                    ans.push_back({nums[i],nums[l],nums[r]});
                    l++,r--;
                   //去重
                    while(l<r&&nums[l]==nums[l-1]) l++;
                    while(l<r&&nums[r]==nums[r+1]) r--;
                }
                else if(nums[l]+nums[r]>-nums[i]) r--;
                else l++;
            }
            //去重
            while(i+1<nums.size()&&nums[i+1]==nums[i]) i++;
        }
        return ans;
    }
};