Codeforces Round 258 (Div. 2) E. Devu and Flowers 生成函数

news2025/3/10 9:57:21

题目链接

题目大意

有 $n$ $(1\leq n \leq 20)$ 个花瓶，第 $i$ 个花瓶里有 $f_i$ $(1\leq f_i \leq 10^{12})$ 朵花。现在要选择 $s$ $(1\leq s \leq 10^{14})$ 朵花。

求出有多少种方案。两种方案不同当且仅当两种方案中至少有一个花瓶选择花的数量不同，答案对 $10^9+7$ 取模。

思路

取第 $i$ 种花的生成函数可以表示为： $f_i(x)=$ $1+x+x^2+x^3+ \cdot \cdot \cdot x^{f_i}$ .

则选取方案的生成函数可以表示为： $G (x) =$ $f_1(x) \cdot f_2(x) \cdot \cdot \cdot f_n(x) =$ $\frac {(1-x^{f_1+1})(1-x^{f_2+1})(1-x^{f_3+1}) \cdot \cdot \cdot (1-x^{f_n+1})} {(1-x)^n}$ $=$ $\frac {\prod_{i=1}^{n}(1-x^{f_i+1})}{(1-x)^{n}}$ $=$ $\prod_{i=1}^{n}(1-x^{f_i+1})$ $\cdot$ $\sum_{i=0}^\infty$ ${i+n-1\choose i}$ $x_i$ $=$ $\prod_{i=1}^{n}(1-x^{f_i+1})$ $\cdot$ $\sum_{i=0}^\infty$ ${i+n-1\choose n-1}$ $x_i$ .

则 $x^s]G(x)$ 即为所求的答案。

令 $A(x)=\prod_{i=1}^{n}(1-x^{f_i+1})$ , $B(x)=\sum_{i=0}^\infty$ ${i+n-1\choose n-1} x_i$ .

由于 $n\leq20$ , 所以 $A (x)$ 最多只有 $2^{20}$ 项，可以直接枚举，即 $ans=[x^s]G(x)=\sum_{i=0}^{2^n}[x^i]A(x) \cdot [x^{s-i}]B(x)=$ $\sum_{i=0}^{2^n}[x^i]A(x) \cdot {s-i+n-1\choose n-1}$ ，由于 $n$ 很小可以暴力计算组合数，总的时间复杂度为 $\cdot 2^n)$ .

code

#include <bits/stdc++.h>
#define int long long
#define ll long long
#define pii pair<int, int>

using namespace std;
const int mod = 1e9 + 7;
int f[30];

int ksm(int x, int k)
{
    int res = 1;
    while (k > 0)
    {
        if (k & 1)
            res = res * x % mod;
        x = x * x % mod;
        k >>= 1;
    }
    return res;
}

signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);
    int n, s;
    cin >> n >> s;
    int fac = 1;
    for (int i = 1; i <= n; ++i)
    {
        cin >> f[i - 1];
        if (i < n)
            fac = fac * i % mod;
    }
    int infac = ksm(fac, mod - 2);
    int ans = 0;
    for (int i = 0; i < (1ll << n); ++i)
    {
        int cnt = 0, sum = 0;
        for (int j = 0; j < n; ++j)
        {
            if ((1ll << j) & i)
                cnt++, sum += f[j] + 1;
        }
        if (sum > s)
            continue;
        int tmp = 1;
        for (int j = 0; j < n - 1; ++j)
        {
            int p = (s - sum + n - 1 - j) % mod;
            tmp = tmp * p % mod;
        }
        tmp = tmp * infac % mod;
        ans = (ans + ((cnt & 1) ? mod - tmp : tmp) % mod) % mod;
    }
    cout << (ans + mod) % mod << '\n';
    return 0;
}

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/2312627.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！