LeetCode15. 3Sum

题目

给一个包含N个整数的数组S，求解是否存在三个元素a, b, c使得a + b + c = 0.

注意：解集中不能包含重复的解[a, b, c]

例如, 数组S = [-1, 0, 1, 2, -1, -4],

解集是：{ [-1, 0, 1], [-1, -1, 2] }

思路

1. 当遇到数a，我们需要寻找两个数的和 b + c = -a，即3Sum变成了2Sum问题如 未排序的Two Sum 和 已排序的Two Sum

2. 为了避免找到重复的解，我们将数组从小到大（有正负）排序，然后左往右扫一遍，只对每个负数 a 处理即可，不用管在 a 左边已经处理过的元素，细节如下：

   • 从 a 的后面开始（下标范围：left = a的下标+1，right = 数组长度-1）寻找 b, c

   • 如果array[left] + array[right] < -a，说明left指向的数不够大，需要left++

   • 如果array[left] + array[right] > -a，说明right指向的数太大，需要right- -

   • 如果array[left] + array[right] = -a，即找到以 a 开头的一组解

     • 避免解重复，left++以向右跳过所有与上面 b 相同的值，right- -以向左跳过与 c 相同的值
   • 重复上面步骤直到left 与 right相遇，即可找出与a相关的解

   上面黑体的两个部分避免了重复解

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {

        vector<vector<int>> result;

        vector<int> item;

        if(nums.size() < 3) return result;

        //按照负到正，小到大排序
        sort(nums.begin(), nums.end());

        for(int i = 0; i < nums.size() - 2; i++) {

            //遍历到正数之后，数列右边再无合适的两个数使三者和为0，终止循环
            if(nums[i] > 0) break;

            //下面把问题转化为twoSum问题，即寻找两数，它们的和为某确定值
            int twoSum = 0 - nums[i];

            //为了避免找到的解与已找到的重复，我们需要从当前位置的下一个位置开始找另两个元素
            int left = i + 1;

            int right = nums.size() - 1;

            while(left < right) {

                //找到一组解
                if (nums[left] + nums[right] == twoSum) {

                    item.push_back(nums[i]);

                    item.push_back(nums[left]);

                    item.push_back(nums[right]);

                    result.push_back(item);

                    item.clear();

                    //跳过重复值
                    while(left < right && nums[left] == nums[left + 1]) left++;

                    while(left < right && nums[right] == nums[right - 1]) right--;

                    left++; right--;

                }
                else if (nums[left] + nums[right] > twoSum) right--;
                else left++;
            }

            while(nums[i] == nums[i + 1]) i++;
        }

        return result;
    }
};