双指针——较小的三数之和（leetcode 259）

题目描述

给定一个长度为 n 的整数数组和一个目标值 target，寻找能够使条件 nums[i] + nums[j] + nums[k] < target 成立的三元组  i, j, k 个数（0 <= i < j < k < n）。

示例：

输入: nums = [-2,0,1,3], target = 2
输出: 2
解释: 因为一共有两个三元组满足累加和小于 2:
     [-2,0,1]
     [-2,0,3]

算法分析

我们用两个指针 left和 right分别指向数组中的第一个和最后一个元素，它们的和为 1+8=9>target，这说明 right 不能出现在二元组中（因为 left只向左移动，如果此时二者的和已经大于 target，那么在 left移动的过程中，二者的和就不可能小于 target了），因此需要将 right 向左移动一位。

代码

class Solution {
public:
    int threeSumSmaller(vector<int>& nums, int target) {
        
        int res = 0;
        std::sort(nums.begin(), nums.end());
        for(int i = 0; i < nums.size(); ++i) {

            int left = i + 1;
            int right = nums.size() - 1;

            while(left < right) {

                if(nums[i] + nums[left] + nums[right] < target) {
                    res += right - left;
                    ++left;
                } else {
                    --right;
                }
            }
        }
        return res;
    }
};

时间复杂度分析

时间复杂度：O(n^2)。在每一步中，要么 left向右移动一位，要么 right向左移动一位。当 left≥right 时循环结束，因此它的时间复杂度为 O(n)。加上外层的循环，总的时间复杂度为 O(n^2)。
空间复杂度：O(1)。