3Sum

2020-02-05 17:56:00 阅读：219 来源： 互联网

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.

Note:

The solution set must not contain duplicate triplets.

Example:

Given array nums = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

给定一个包含n个整数的数组nums，判断该数组中是否存在整数a、b、c，使它们满足a+b+c=0？找出所有满足该条件的整数。
注意：找出的3个数不可完全相同。

分析：
为求三个数的和为0，可先固定一个数ai，再在数组中找其余两个数aj、ak，使它们满足aj+ak=-ai。
为加快查找速度，可先对数组排序，这样当遍历该数组时，当遍历到第i个元素时，若ai大于0，可提前终止遍历，因为后面的数只可能是大于或等于它，ai+aj+ak必大于0。若ai小于等于0，从ai+1到an-1中查找满足条件的数。代码如下：

    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> rst;
        int cnt = nums.size();
        if (cnt < 3)
            return rst;
        int target = 0, left = 0, right = 0, tmp = 0;
        sort(nums.begin(), nums.end());
        if (nums[0] > 0 || nums[cnt - 1] < 0)
            return rst;
        for (int i = 0; i < cnt - 2; i++)
        {
            if (nums[i] > 0)
                break;
            if (i > 0 && nums[i] == nums[i - 1])
                continue;
            target = 0 - nums[i];
            left = i + 1;
            right = cnt - 1;
            while (left < right)
            {
                tmp = nums[left] + nums[right];
                if (tmp == target)
                {
                    rst.push_back({ nums[i],nums[left],nums[right] });
                    while (left < right)
                    {
                        if (nums[left] != nums[left + 1])
                            break;
                        left++;
                    }
                    while (left < right)
                    {
                        if (nums[right] != nums[right - 1])
                            break;
                        right--;
                    }
                    left++;
                    right--;
                }
                else if (tmp < target)
                    left++;
                else
                    right--;
            }
        }
        return rst;
    }

标签：cnt,right,nums,3Sum,ai,rst,left
来源： https://www.cnblogs.com/big-potato/p/12264377.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

3Sum