LeeCode 1498 二分 + 快速幂

最新推荐文章于 2023-05-09 16:46:53 发布

SHOHOKUKU

最新推荐文章于 2023-05-09 16:46:53 发布

阅读量5.1k

点赞数

分类专栏：数学基本算法

本文链接：https://blog.csdn.net/neweryyy/article/details/107003510

版权

基本算法同时被 2 个专栏收录

252 篇文章 1 订阅

订阅专栏

数学

164 篇文章 0 订阅

订阅专栏

题意

传送门 LeeCode 1498. 满足条件的子序列数目

题解

求 $n u m s$ 中能满足其最小元素与最大元素的和小于或等于 $t a r g e t$ 的非空子序列的数目，则答案与原数组顺序无关。

排序 $n u m s$ ，遍历数组，二分求以当前元素 $x$ 为最小元素的子序列的和满足条件时，序列最大元素元素的 $upper\_bound(target-x)$ ，若该值小于等于当前元素，则不存在这样的子序列；反之，考虑以 $y(x\leq y<upper\_bound(target)$ 为最大值的子序列，对于序列中的其他元素 $z (x < z < y)$ 可以取与不取，设 $[x, y]$ 的元素数量为 $n^{'}$ ，则子序列数量为 $2^{n'-2}$ ；设 $x, upper\_bound(target))$ 的元素数量为 $n$ ，则以 $x$ 为最小元素的子序列数量为

$2^{n-2}+2^{n-3}+\dots +2^0$

式中最后一项为 $n = 2$ ，即子序列只有 $2$ 个元素的情况；考虑子序列数量为 $1$ ，即子序列中仅有 $x$ ，则满足条件的以 $x$ 为最小元素的子序列数为

$2^{n-1}-1+1=2^{n-1}$

快速幂求解即可。

class Solution
{
public:
    int mod = 1000000007;
    int qpow(long long x, int n)
    {
        if (n < 0) return 0;
        long long res = 1;
        while (n)
        {
            if (n & 1) res = res * x % mod;
            x = x * x % mod;
            n >>= 1;
        }
        return res;
    }
    int numSubseq(vector<int> &nums, int target)
    {
        int res = 0;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size(); i++)
        {
            int n = upper_bound(nums.begin(), nums.end(), target - nums[i]) - (nums.begin() + i);
            res = (res + qpow(2, n - 1)) % mod;
        }
        return res;
    }
};