BZOJ3157/BZOJ3516/BZOJ4126：国王奇遇记（DP）

最新推荐文章于 2019-08-31 23:02:00 发布

DZYO

最新推荐文章于 2019-08-31 23:02:00 发布

阅读量724

点赞数

分类专栏： DP及DP优化

本文链接：https://blog.csdn.net/qq_35649707/article/details/79587820

版权

DP及DP优化专栏收录该内容

70 篇文章 0 订阅

订阅专栏

传送门

题解：
虽然我知道这只是弱化版，不过其他两道太神了看不懂，而且今天学的扩展板也还没有消化，就先水一水这道题吧。

观察这个式子： $f_{n,m}=\sum\limits_{i=1}^n i^m m^i$

$n$ 远远大于 $m$ ，不能直接枚举。
那么怎么办？递推！
$f_{n,m}$ 的大部分信息在前面已经计算出来过，我们可以想办法构造递推式。不妨记 $f_{n,j} = \sum\limits_{i=0}^n i^jm^i (j \not = 0)$ 。考虑从 $f_{n,k}(k\lt j)$ 的递推：

f n, j = = = = = \sum i = 0 n i j m i m \sum i = 0 n - 1 (i + 1) j m i m \sum i = 0 n - 1 m i \sum k = 0 j (j k) i k m \sum k = 0 j (j k) f n - 1, k m \sum k = 0 j (j k) (f n, k - n k m n)

$\begin{matrix} f_{n,j}&=&\sum\limits_{i=0}^n i^jm^i \\ &=&m \sum\limits_{i=0}^{n-1} (i+1)^jm^i \\ &=&m \sum\limits_{i=0}^{n-1}m^i\sum\limits_{k=0}^{j}\binom{j}{k}i^k \\ &=&m\sum\limits_{k=0}^{j}\binom{j}{k}f_{n-1,k} \\ &=&m\sum\limits_{k=0}^{j}\binom{j}{k}(f_{n,k}-n^km^n) \end{matrix}$

注意第二步的转移虽然看起来很莫名其妙，其实是利用组合数构造了这一个递推式子。

$O(m^2)$ 做动态规划即可。
（可以利用多项式插值或者快速傅里叶变换求解 $\sum\limits_{i=0}^n f_im^i$ 且 $f_i$ 是 $k(k \le 5e5)$ 次多项式的问题，以后会把这个坑填上）。

upt :已经填坑，具体请看:http://blog.csdn.net/qq_35649707/article/details/79595192

#include <bits/stdc++.h>
using namespace std;
typedef long long LL; 
const int N=2e2+50;
const int mod=1e9+7;
inline int mul(int x,int y) {return (LL)x*y%mod;}
inline int add(int x,int y) {return (x+y>=mod)?(x+y-mod):x+y;}
inline int dec(int x,int y) {return (x-y<0)?(x-y+mod):x-y;}
inline int power(int a,int b) {
    int rs=1;
    for(;b;b>>=1,a=mul(a,a)) if(b&1) rs=mul(rs,a);
    return rs;
}
int n,m,a,inv,c[N][N],s[N],pw[N],f[N];
int main() {
    cin>>n>>m; pw[0]=1; a=power(m,n); 
    inv=power(mod+1-m,mod-2);
    if(m==1) {printf("%d\n",(LL)n*(n+1)/2%mod); return 0;}
    for(int i=0;i<=m;i++) c[i][0]=1;
    for(int i=1;i<=m;i++) pw[i]=mul(pw[i-1],n);
    for(int i=1;i<=m;i++) for(int j=1;j<=i;j++) c[i][j]=add(c[i-1][j-1],c[i-1][j]);
    f[0]=mul(dec(power(m,n+1),1),power(m-1,mod-2));
    for(int i=1;i<=m;i++) {
        int rs=mod-mul(pw[i],a);
        for(int j=0;j<i;j++) rs=add(rs,mul(c[i][j],dec(f[j],(LL)pw[j]*a%mod)));
        rs=mul(rs,m); rs=mul(rs,inv); f[i]=rs; 
    }
    printf("%d\n",f[m]);
}