leetcode 15. 三数之和

15. 三数之和

问题描述

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

问题分析

该问题实际上把一个三数之和问题转化为了两数之和问题，特别要注意的是要注意去重，以及判断条件 while(left < right)不要写为 while(left <= right)。

代码如下：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        //vector<int> a;
        int n = nums.size();
        //if(n == 0) return ans;
        //sort(&nums[0],&nums[0]+n);
        sort(nums.begin(),nums.end());
        int right,left,i,tmp;
        for(i = 0; i < n; i++)
        {
            tmp = nums[i];
            if(tmp > 0) break;//因为已经排好序了，后面的值肯定比tmp还要大，三个数的和一定大于0
            if((i > 0 && nums[i] == nums[i-1])) continue; //去除重复元素
            left = i + 1;right = n-1;
            //while(left <= right)//注意不要写为小于等于号
            while(left < right)
            {
                if(nums[left]+nums[right]+tmp<0)left++;//和太小，左端的数需要增加
                else if(nums[left]+nums[right]+tmp>0)right--;//和太大，右端的数需要减小
                else{
                    ans.push_back({tmp,nums[left],nums[right]});
                    left++;
                    right--;
                    while (left < right && nums[left] == nums[left - 1]) ++left;
                    while (left < right && nums[right] == nums[right + 1]) --right;
                }
            }
        }
        return ans;
    }
};

注意

在排序时，如果我们把
sort(nums.begin(),nums.end());
替换为
if(n == 0) return ans;
sort(&nums[0],&nums[0]+n);//注意不是sort(&nums[0],&nums[0]+n-1);

效率会有不同.