LeetCode15 三数之和

题目描述

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

思路

求解三数之和，采用双指针的思路：

1. 首先对数组按照从小到大进行排序
2. 对于数组中的每个值nums[i]：令左侧指针下表为 left = i + 1，右侧指针 right = n - 1。如果nums[i] + nums[left] + nums[right] 大于0，则减小right让和减小；大于0，则增大left让和增加；等于零则将其加入答案中。
3. 要注意数组中相同元素可能让答案重复输出

代码

class Solution {
public:
    static bool cmp(int a, int b){
        return a < b;
        
    }
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        int n = nums.size();
        sort(nums.begin(), nums.end(), cmp);
        for(int i = 0; i+1 < n; i++){
            if(nums[i] > 0) break;
            if(i > 0 && nums[i] == nums[i-1]) continue;
            int left = i+1;
            int right = n -1;
            while(left < right){
                if(nums[i] + nums[left] + nums[right] > 0){
                    right--;
                }

                else if(nums[i] + nums[left] + nums[right] < 0){
                    left++;
                }else{
                    vector<int> a;
                    a.push_back(nums[i]);
                    a.push_back(nums[left]);
                    a.push_back(nums[right]);
                    ans.push_back(a);
                    left++;
                    right--;
                    while(left<right && nums[left -1] == nums[left]){
                        left++;
                    }
                    while(left<right && nums[right] == nums[right+1]){
                        right--;
                    }
                }
            }
        }
        return ans;
    }
};