[LeetCode]15. 3Sum

Problem

Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

The solution set must not contain duplicate triplets.

Example:

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

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

给定一个数组, 在数组中查找元素a,b,c, 三个数满足a+b+c = 0; 找到所有符合条件的解. 但是不能包含重复解.

Solution

Analysis

因为数组中可能存在重复元素,导致解也有可能重复. 所以我们先对数组进行排序, 数组排序后,相等元素在位置上相邻. 所以我们可以对重复数字进行跳过.

问题的求解. 数组经过排序, 数组有序,然后固定一个元素, 再对剩下区间元素使用双指针进行遍历,查找. 如果找到后, 同时变更指针, 让它们跳过重复元素[会生成重复解]. 然后继续遍历. 循环条件是left < right.

3Sum, 固定一个数, 然后使用两个指针进行遍历查找,时间复杂度O(N * logN).

方法总结就是: 双指针+夹逼.

K-Sum: 外层k-2次循环, 内部使用双指针进行夹逼.

Code

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        if (nums.size() < 3) return result;
        
        sort(nums.begin(), nums.end());
        
        
        for (int i=0; i< nums.size(); i++){
            if(i > 0 && nums[i] == nums[i-1]) continue;// 跳过固定元素
            
            int left = i+1, right = nums.size()-1;
            
            while (left < right){
                int sum = nums[i] + nums[left] + nums[right];
                
                if (sum < 0)
                    left++;
                else if (sum > 0)
                    right--;
                else{
                    result.push_back(vector<int>{nums[i], nums[left], nums[right]});
                    left++, right--;// 先跳过这个解
                    // 如果接下来的数和之前的数相同, 跳过
                    // 先进行有效性判断, left< right; 否则出现下标越界
                    while (left < right && nums[left] == nums[left-1]) left++;// 跳过重复元素解
                    while (left < right && nums[right] == nums[right+1]) right--;
                    
                }
            }
        }
        
        return result;
    }
};

Problem

Solution

Analysis

Code

推荐文章（由hexo文章推荐插件驱动）