LeetCode #259 - 3Sum Smaller

最新推荐文章于 2024-06-04 16:45:45 发布

LawFile

最新推荐文章于 2024-06-04 16:45:45 发布

阅读量164

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/LawFile/article/details/86663120

版权

LeetCode 专栏收录该内容

625 篇文章 1 订阅

订阅专栏

题目描述：

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.

Example:

Input: nums = [-2,0,1,3], and target = 2

Output: 2

Explanation: 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(n^2) runtime?

class Solution {
public:
    int threeSumSmaller(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        int result=0;
        int n=nums.size();
        for(int i=0;i<n-2;i++)
        {
            int j=i+1;
            int k=n-1;
            while(j<k)
            {
                if(nums[i]+nums[j]+nums[k]<target)
                {
                    result+=k-j;
                    j++;
                }
                else k--;
            }
        }
        return result;
    }
};