Leetcode 015. 3Sum 三个整数的和 二分查找

题目链接

题目描述

给出一个 nums 数组，找到其中的三个数 a,b,c 使得 a+b+c=0，找到所有满足这样条件的不重复的三元组返回。

示例：

Given array nums = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

解题思想

这个题目是 2Sum 的升级版，需要在数组中寻找三个数和为一个给定的数，只能把问题拆分开，按顺序遍历数组，有了一个数以后，我们去找和为 0 减去对应数字的二元组。找二元组的话，可以先将整个数组排序，然后利用双指针法，两个指针分别从左边和后边开始走，如果满足条件，加入到结果中。

这里还需要注意一点，就是为了不保证重复，两个指针在走的过程中，如果碰到了重复的数，需要继续一直往后走。

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        if (nums.size() < 3) return {};
        sort(nums.begin(), nums.end());
        vector<vector<int>> result;
        for (int i = 0; i < nums.size() - 2; i ++){

            if (i != 0 && nums[i] == nums[i - 1])
                continue;

            int sum = 0 - nums[i];

            int left = i + 1, right = nums.size() - 1;
            while(left < right){
                if (nums[left] + nums[right] == sum){
                    result.push_back({nums[i], nums[left], nums[right]});
                    while(left < right && nums[left + 1] == nums[left]) left ++;
                    while(left < right && nums[right - 1] == nums[right]) right --;
                    right --;
                    left ++;
                }
                else if (nums[left] + nums[right] < sum){
                    left ++;
                }
                else {
                    right --;
                }
            }
        }
        return result;
    }
};