3sum来自LeetCode

最新推荐文章于 2025-09-12 17:31:31 发布

转载最新推荐文章于 2025-09-12 17:31:31 发布 · 55 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/jianbo1995/p/10770901.html

文章标签：

#数据结构与算法

本文深入探讨了解决三数之和为零问题的高效算法。通过对输入向量进行排序，利用双指针技巧，避免重复组合，实现O(N^2)时间复杂度的解决方案。文章详细介绍了如何处理重复元素，确保结果的唯一性。

题目给定一个输入vector，要求找出任意三个数和为0的任意组合，组合不可重复。

这个题目基本也就只想到前面的排序，看了一个submit，整理了一下思路：首先排序，然后通过一个O（N^2）的遍历，加上一个尾指针，完成对所有数字的可能组合的判断

但是在"指针"移动的过程中需要考虑到重复数字，即找到一个组合之后j、k需要到达不重复的新位置，以及i到下一次循环的时候也要避开重复数字，这里踩了坑，看了第二遍

代码才发现自己的问题。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        
        vector<vector<int>> result;
        if(nums.empty())
            return result;
       
        sort(nums.begin(),nums.end());
        int sum;
        int j,k;
        
        for(int i=0;i<nums.size()-1;i++)
        {
            if(i>0&&nums[i]==nums[i-1])
                continue;
            j=i+1;
            k=nums.size()-1;
            
            while(j<k)
            {
                sum = nums[i] + nums[j] + nums[k];
                if(sum > 0)
                    k--;
                else if(sum < 0)
                    j++;
                else
                {
                    result.push_back(vector<int>{nums[i],nums[j],nums[k]});
                    while(j<k-1&&nums[j]==nums[j+1])
                        j++;
                    while(k>j&&nums[k]==nums[k-1])
                        k--;
                    j++;
                    k--;
                }
            }
        }
        return result;
    }
};