zoukankan      html  css  js  c++  java
  • 【集训队作业2018】复读机

    即使是一道菜题,也掩盖不了窝生成函数弱的事实。

    窝看到题目只想到了DP,还是生成函数不够熟。然后直到神仙WWJ提醒我用生成函数……

    首先是排列,那就可以写成指数型生成函数的优美形式。直接表示成exp吧,然后d的倍数就用单位根性质来处理。

    所以实际上就是求 ([x^n] (frac{sum_{i=0}^{D - 1} e^{omega_{D}^{i} x}}{D})^K)

    显然可以把底下的 (D^K) 提出来,此时求上面那个东西。

    怎么求第 (n) 项系数啊?

    这时候可以发现 (D) 大的时候 (K) 小了(然后只看了最后一条范围就被坑了)

    于是发现可以暴力二项式展开!

    枚举 (i) 次单位根的个数,复杂度 (O(K^{D - 1}))

    然后剩下的东西根据指数函数的性质可以合并成一项 (e^{Cx}) ,此时系数只是一个快速幂的事情了。

    单位根的事情就随便求一个丢上去好了。

    #include <bits/stdc++.h>
    
    const int mod = 19491001;
    const int MAXN = 500010;
    typedef long long LL;
    int mul(int a, int b) { return (LL) a * b % mod; }
    void reduce(int & x) { x += x >> 31 & mod; }
    int fastpow(int a, int b) {
    	int res = 1;
    	for (; b; b >>= 1, a = mul(a, a)) if (b & 1) res = mul(res, a);
    	return res;
    }
    int n, K, D, fac[MAXN], inv[MAXN];
    int C(int a, int b) { return (LL) fac[a] * inv[b] % mod * inv[a - b] % mod; }
    int pr[3], ans;
    void dfs(int S, int rest, int sum, int way) {
    	if (S == D - 1) {
    		reduce(sum += mul(pr[S], rest) - mod);
    		reduce(ans += mul(fastpow(sum, n), way) - mod);
    		return ;
    	}
    	for (int i = 0; i <= rest; ++i)
    		dfs(S + 1, rest - i, (sum + (LL) pr[S] * i) % mod, mul(way, C(rest, i)));
    }
    int main() {
    	std::cin >> n >> K >> D;
    	fac[0] = fac[1] = inv[0] = inv[1] = 1;
    	for (int i = 2; i <= K; ++i) {
    		fac[i] = mul(fac[i - 1], i);
    		inv[i] = mul(inv[mod % i], mod - mod / i);
    	}
    	for (int i = 2; i <= K; ++i)
    		inv[i] = mul(inv[i - 1], inv[i]);
    	pr[0] = 1;
    	if (D == 2) pr[1] = mod - 1;
    	if (D == 3) pr[1] = 663067, pr[2] = mul(pr[1], pr[1]);
    	dfs(0, K, 0, 1);
    	ans = mul(ans, fastpow(D, mod - 1 - K));
    	std::cout << ans << std::endl;
    	return 0;
    }
    
  • 相关阅读:
    局部变量、全局变量和修改全局变量
    python中函数的参数
    python之匿名函数和递归函数
    设计模式之职责链模式
    设计模式之代理模式
    设计模式之flyweight享元模式
    设计模式之外观模式
    设计模式之装饰模式
    组合模式更清晰的例子
    设计模式之组合模式
  • 原文地址:https://www.cnblogs.com/daklqw/p/11588389.html
Copyright © 2011-2022 走看看