[Leetcode] 259. 3Sum Smaller 解题报告

题目：

Given an array of n integers nums and a target, find the number of index triplets i, j, k with 0 <= i < j < k < n that satisfy the condition nums[i] + nums[j] + nums[k] < target.

For example, given nums = [-2, 0, 1, 3], and target = 2.

Return 2. Because there are two triplets which sums are less than 2:

[-2, 0, 1]
[-2, 0, 3]

Follow up:
Could you solve it in O(n2) runtime?

思路：

由于本题目是求三个数的和，并且不要求返回索引位置（只要求返回符合条件的triple个数），所以可以首先花费O(nlogn)的时间复杂度进行排序。为了将时间复杂度降低到O(n^2)，我们采用一个技巧，就是利用start的递增性和end的递减性。例如对于start = i + 1，假设我们找到了一个合适的end，使得i和[start, end]之间的数构成的triple都满足条件，那么对于start = i + 2, ... nums.size() - 2，其对应的end必然是逐渐递减的。因此，我们不需要对每个start都用二分法查找合适的end，只需要在第二重循环中，要么增加start，要么减少end（如果用普通的思路做，则算法的时间复杂度是O(n^2logn)）。

代码：

class Solution {
public:
    int threeSumSmaller(vector<int>& nums, int target) {
        if(nums.size() < 3) {
            return 0;
        }
        sort(nums.begin(), nums.end());
        int ret = 0, sum = 0;
        for(int i = 0; i < nums.size() - 2; ++i) {
            int start = i + 1, end = nums.size() - 1;
            while(start < end) {
                sum = nums[i] + nums[start] + nums[end];
                if(sum < target) {
                    ret += (end - start++);     // all the ones satisfy the requirement
                }
                else {
                    --end;
                }
            }
        }
        return ret;
    }
};