3Sums

15. 3Sums

• 题目链接
• 题意
• 解法
• 代码
• 算法复杂度
• 要点

题目链接：15. 3Sums

题意

给出一个数组，求出该数组中所有和为给定值的三元组的集合

解法

在有序数组中可以从用两个端点开始的两个下标来逐步逼近目标和，因此可以先将给定数组排序，然后每次固定一个数，在剩下的区间用左右两个下标来逼近给定和。

代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int len = nums.size();
        int left, right, current, temp;
        sort(nums.begin(), nums.end());//sort the array
        vector<vector<int>> result;
        if (len < 3 || 3*nums[0]>0||3*nums[len-1]<0)
            return result;


        for (current = 0; current <= len - 3; ++current) {
            if (3 * nums[current] > 0) {//too large
                continue;
            }
            else if (nums[current]+2*nums[len-1]<0) {//too small
                continue;
            }
            left = current + 1;
            right = len - 1;
            while (left < right) {
                if (nums[current]+nums[left]+nums[right] < 0) {
                    ++left;
                }
                else if (nums[current] + nums[left] + nums[right] > 0) {
                    --right;
                }
                else {
                    result.push_back({ nums[current],nums[left],nums[right] });
                    temp = nums[left];
                    while (left < right&&nums[left] == temp)
                        ++left;
                    temp = right;
                    while (left < right&&nums[right] == temp)
                        --right;
                }
            }
            while (current <= len - 3 && nums[current + 1] == nums[current]) 
                ++current;
        }
        return result;
    }
};

算法时间复杂度：O(n^2)

要点

• 想要使用夹逼的方法来逐步逼近给定值，需要先将数组排序，可以使用algorithm头文件里面的sort()方法
• 每次固定第一个数，剩下的两个数与当前固定数之和如果大于给定值，就将右边界左移，反之则将左边界右移