[CF995F]Cowmpany Cowmpensation

codeforces

description

一棵\(n\)个节点的树，给每个节点标一个\([1,m]\)之间的编号，要求儿子的权值不大于父亲权值。求方案数。\(n\le3000,n\le10^9\)

sol

可以证明答案是关于\(m\)的一个\(n\)次多项式。我不会证。
如果\(P(x)\)是关于\(x\)的\(n\)次多项式，则有
\[P(x)=\sum_{i=0}^{n}(-1)^{n-i}P(i)\frac{x(x-1)...(x-n)}{(n-i)!i!(x-i)}\]
可见杜教\(\mbox{PPT}\)《多项式与求和》。
所以只要对\([1,n]\)求答案就可以了，很显然是一个\(O(n^2)\)的\(dp\)，所以复杂度是\(O(n^2)\)。

code

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int gi(){
    int x=0,w=1;char ch=getchar();
    while ((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
    if (ch=='-') w=0,ch=getchar();
    while (ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return w?x:-x;
}
const int N = 3005;
const int mod = 1e9+7;
int n,m,nxt[N],head[N],f[N][N],inv[N],ans;
void dfs(int u){
    for (int i=1;i<=n;++i) f[u][i]=1;
    for (int v=head[u];v;v=nxt[v]){
        dfs(v);
        for (int i=1;i<=n;++i)
            f[u][i]=1ll*f[u][i]*f[v][i]%mod;
    }
    for (int i=2;i<=n;++i) (f[u][i]+=f[u][i-1])%=mod;
}
int main(){
    n=gi();m=gi();inv[0]=inv[1]=1;
    for (int i=2;i<=n;++i) inv[i]=1ll*inv[mod%i]*(mod-mod/i)%mod;
    for (int i=2,ff;i<=n;++i)
        nxt[i]=head[ff=gi()],head[ff]=i;
    dfs(1);
    if (m<=n) return printf("%d\n",f[1][m]),0;
    for (int i=1;i<=n;++i){
        int sum=f[1][i];
        for (int j=0;j<=n;++j)
            if (j!=i) sum=1ll*sum*(m-j)%mod*(i>j?inv[i-j]:mod-inv[j-i])%mod;
        (ans+=sum)%=mod;
    }
    printf("%d\n",ans);return 0;
}

转载于:https://www.cnblogs.com/zhoushuyu/p/9387662.html