Leetcode 15.三数之和（中级算法-数组）

题目描述

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
 [-1, 0, 1],
 [-1, -1, 2]
]

解题历程

1、第一次通过时的思路

a和b确定了c就可以唯一确定了，也就是两个元素确定一个解，先排个序，然后用两层循环确定 i 和 j ，然后二分搜索第三个值即可。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        
        vector<vector<int>> ans;
        if(nums.size()<3)return ans;    //数组长度小于3时无解，直接返回空ans
        vector<int>tmp;
        sort(nums.begin(),nums.end());  //对数组排序，便于搜索，nlogn
                
        for(int i=0;i<nums.size()-2;i++)    //确定两个数后，第三个数即确定，直接用搜索  (n^2)*logn
        {
            for(int j=i+1;j<nums.size()-1;j++)
            {
                int m,left=j+1,right=nums.size()-1,x=-nums[i]-nums[j];
                while(left<=right)
                {
                    m=(left+right)/2;            //对半搜索
                    if(x>nums[m])left=m+1;
                    else if(x<nums[m])right=m-1;
                    else break;
                }
                
                if(left<=right) //left小于等于right说明搜索成功
                {
                    tmp.push_back(nums[i]);
                    tmp.push_back(nums[j]);
                    tmp.push_back(nums[m]);
                    ans.push_back(tmp);     //将结果添加到解向量
                    tmp.clear();
                }
                for(;j+1<nums.size()&&nums[j]==nums[j+1];j++);   //当两个数确定后，这组解已经是唯一的，跳过重复的（已排序容易跳过）
            }
            for(;i+1<nums.size()&&nums[i]==nums[i+1];i++);  //跳过重复
        }
        return ans;
    }
};

2、更优的解答

可以用双指针，耗时更短，待更。。。