Leetcode 15: 3 sum

最新推荐文章于 2019-08-03 17:11:38 发布

原创最新推荐文章于 2019-08-03 17:11:38 发布 · 97 阅读

0 ·

CC 4.0 BY-SA版权

算法题专栏收录该内容

118 篇文章

订阅专栏

本文探讨了如何在一个整数数组中寻找三个元素，使它们的和等于零，并提供了一个有效的C++解决方案。该算法首先对数组进行排序，然后使用双指针技术遍历数组以找到所有不重复的三元组。

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

Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

For example, given array S = {-1 0 1 2 -1 -4},
A solution set is:
(-1, 0, 1)
(-1, -1, 2)

C++
 vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        std::sort(nums.begin(),nums.end());
        for(int k = 0;k < nums.size();++k)
        {
            if(nums[k] > 0)break;
            if(k > 0&&nums[k] == nums[k - 1])continue;
            int i = k + 1,j = nums.size() - 1;
            while(i < j)
            {
                if(nums[k] + nums[i] + nums[j] == 0)
                {
                    res.push_back({nums[k],nums[i],nums[j]});
                    while(i < j&&nums[i] == nums[i + 1])++i;
                    while(i < j&&nums[j] == nums[j - 1])--j;
                    ++i;
                    --j;
                }
                else if(nums[k] + nums[i] + nums[j] > 0)
                    --j;
                else
                    ++i;
            }
        }
        return res;
    }