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]
]

---

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        int flag=0;
        sort(nums.begin(),nums.end());
        if (nums.size()<3) return result;
        for (int i =0;i<nums.size()-2;i++ )
        {
            if (i>0&&nums[i]==nums[i-1]) continue;
            flag =0;
            for (int j =i+1;j<nums.size()-1;j++ )
            {
                if (j>1&&nums[j]==nums[j-1]&&(flag==1))  continue;
                vector<int> temp;
                for (int m =j+1;m<nums.size();m++ )
               {
                   if(nums[i]+nums[j]+nums[m]==0)
                   {
                            temp.push_back(nums[i]);
                            temp.push_back(nums[j]);
                            temp.push_back(nums[m]);
                            result.push_back(temp);
                            flag = 1;
                            break;
                            
                   }
               }
            }
        }
        return result;
        
    }
};

超时！

class Solution
{
public:
    vector<vector<int> > threeSum(vector<int> &num)
    {
        vector<vector<int>> result;
        int len = num.size();
        if(len < 3)
            return result;
        sort(num.begin(),num.end());
        for(int i = 0; i < len; i++)
        {
            if (nums[i]>0) break;//因为排好序了，所以第一个数不能大于0
            int j = i + 1;
            int k = len - 1;
            if(i != 0 && num[i] == num[i - 1])//如果遇到重复元素的情况，避免多次考虑
                continue;
            while(j < k)//对于每一个num[i]从i之后的元素中,寻找对否存在三者之和为0的情况
            {
                if(num[i] + num[j] +num[k] == 0)//当三者之和为0的情况
                {
                    result.push_back({num[i],num[j],num[k]});
                    j++;//当此处的j,k满足时，别忘了向前/向后移动，判断下一个是否也满足
                    k--;
                    while(j < k && num[j] == num[j - 1])//如果遇到j重复的情况，也要避免重复考虑
                        j++;
                    while(j < k && num[k] == num[k + 1])//如果遇到k重复的情况，也要避免重复考虑
                        k--;
                }
                else if(num[i] + num[j] + num[k] < 0)//三者之和小于0的情况，说明num[j]太小了，需要向后移动
                    j++;
                else//三者之和大于0的情况，说明num[k]太大了，需要向前移动
                   k--;
            }
        }
        return result;
    } 
}; 

这个很好理解，外面是大循环，里面是两个指针，分别指向首尾，每次判断三个数的和和0的大小关系来决定指针如何移动。