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

简单分析

只能想到最暴力的解法，然后O(n^3)肯定是不行的，去参考了一下题解，发现一种优化是用排序+双指针，这就意味着我们需要给排序后的元素找凑零的两个解，而且要求不重复，那么大于0的元素可以把不考虑了，因为他们本身肯定已经被前面的组合了，只考虑小于等于0的元素，这时可以设置双指针，一个从左向右遍历，一个从右向左，左+右+当前>0，说明右边的过大，右指针向左滑动，小于0左指针向右滑动，去重的话即为相等元素直接跳过，若左右相遇，且仍未找到=0，则本次指针遍历结束，为下一元素重复开始遍历。

代码

class Solution {
public:
   vector<vector<int>> threeSum(vector<int>& nums) {
    vector<vector<int>> res;
    int len = nums.size();
    if(len<3) return res;
    sort(nums.begin(),nums.end());
    if(nums[0]>0||nums[len-1]<0) return res;
    for(int i=0;i<len-1;i++){
        if(nums[i]>0)
            break;
        int target = 0 - nums[i];
        int left = i+1;
        int right = len-1;
        while(left < right){
            if(nums[right] < 0)
                break;
            if(nums[left] + nums[right] < target){
                int v= nums[left];
                while(left != right && nums[left] == v)
                    left++;
            }
            else if(nums[left] + nums[right] > target){
                int v= nums[right];
                while(left != right && nums[right] == v)
                    right --;
            }else{
                vector<int> tmp_res{nums[i],nums[left],nums[right]};
                res.push_back(tmp_res);
                int v= nums[left];
                while(left != right &&  nums[left] == v)
                    left++;
                v= nums[right];
                while(left != right &&  nums[right] == v)
                    right --;
            }
        }
        while(i<len-1&&nums[i] == nums[i+1]) i++;
    }
    return res;
    }
};