P4827 [国家集训队] Crash 的文明世界(第二类斯特林数+树形dp)

传送门

对于点\(u\),所求为\[\sum_{i=1}^ndis(i,u)^k\]
把后面那堆东西化成第二类斯特林数,有\[\sum_{i=1}^n\sum_{j=0}^kS(k,j)\times j!\times{dis(i,u)\choose j}\]
\[\sum_{j=1}^nS(k,j)\times j!\sum_{i=0}^k{dis(i,u)\choose j}\]
于是对于每个点只要维护好\(\sum_{i=0}^k{dis(i,u)\choose j}\)就好了

因为\({n\choose m}={n-1\choose m-1}+{n-1\choose m}\),所以有\[\sum_{i=0}^k{dis(i,u)\choose j}=\sum_{i=0}^k{dis(i,u)-1\choose j-1}+\sum_{i=0}^k{dis(i,u)-1\choose j}\]
那么就可以用树形dp来转移了,设\(dp[u][i]=\sum_{j=0}^k{dis(j,u)\choose i}\),则子树中的转移为\[dp[u][i]=\sum_{v\in son[u]}dp[v][i]+dp[v][i-1]\]
然后子树转移完之后还需要再从父亲转移一次,那么就是两次dfs的事。最后对于每个点直接计算就行了

//minamoto
#include<bits/stdc++.h>
#define R register
#define fp(i,a,b) for(R int i=a,I=b+1;i<I;++i)
#define fd(i,a,b) for(R int i=a,I=b-1;i>I;--i)
#define go(u) for(int i=head[u],v=e[i].v;i;i=e[i].nx,v=e[i].v)
using namespace std;
char buf[1<<21],*p1=buf,*p2=buf;
inline char getc(){return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;}
int read(){
    R int res,f=1;R char ch;
    while((ch=getc())>'9'||ch<'0')(ch=='-')&&(f=-1);
    for(res=ch-'0';(ch=getc())>='0'&&ch<='9';res=res*10+ch-'0');
    return res*f;
}
char sr[1<<21],z[20];int C=-1,Z=0;
inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
void print(R int x){
    if(C>1<<20)Ot();if(x<0)sr[++C]='-',x=-x;
    while(z[++Z]=x%10+48,x/=10);
    while(sr[++C]=z[Z],--Z);sr[++C]='\n';
}
const int N=5e4+5,K=155,P=10007;
inline int add(R int x,R int y){return x+y>=P?x+y-P:x+y;}
inline int dec(R int x,R int y){return x-y<0?x-y+P:x-y;}
inline int mul(R int x,R int y){return x*y-x*y/P*P;}
int ksm(R int x,R int y){
    R int res=1;
    for(;y;y>>=1,x=mul(x,x))if(y&1)res=mul(res,x);
    return res;
}
struct eg{int v,nx;}e[N<<1];int head[N],tot;
inline void add_edge(R int u,R int v){e[++tot]={v,head[u]},head[u]=tot;}
int dp[N][K],now[K],s[K][K],fac[N];
int n,k,res,u,v;
void init(){
    fac[0]=1;fp(i,1,k)fac[i]=mul(fac[i-1],i);
    s[0][0]=1;fp(i,1,k)s[i][i]=s[i][1]=1;
    fp(i,1,k)fp(j,1,i-1)s[i][j]=add(mul(s[i-1][j],j),s[i-1][j-1]);
}
void dfs1(int u,int fa){
    dp[u][0]=1;
    go(u)if(v!=fa){
        dfs1(v,u);dp[u][0]=add(dp[u][0],dp[v][0]);
        fp(j,1,k)dp[u][j]=add(dp[u][j],add(dp[v][j-1],dp[v][j]));
    }
}
void dfs2(int u,int fa){
    go(u)if(v!=fa){
        now[0]=dp[u][0]-dp[v][0];
        fp(j,1,k)now[j]=dec(dp[u][j],add(dp[v][j],dp[v][j-1]));
        dp[v][0]=add(dp[v][0],now[0]);
        fp(j,1,k)dp[v][j]=add(dp[v][j],add(now[j],now[j-1]));
        dfs2(v,u);
    }
}
int main(){
//  freopen("testdata.in","r",stdin);
    n=read(),k=read(),init();
    fp(i,1,n-1)u=read(),v=read(),add_edge(u,v),add_edge(v,u);
    dfs1(1,0),dfs2(1,0);
    fp(i,1,n){
        res=0;
        fp(j,1,k)res=add(res,mul(s[k][j],mul(fac[j],dp[i][j])));
        print(res);
    }return Ot(),0;
}

原文地址:https://www.cnblogs.com/bztMinamoto/p/10220491.html

时间: 2024-10-12 16:18:23

P4827 [国家集训队] Crash 的文明世界(第二类斯特林数+树形dp)的相关文章

bzoj 2159 Crash 的文明世界 —— 第二类斯特林数+树形DP

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=2159 使用公式:\( n^{k} = \sum\limits_{i=0}^{k} S(k,i) * i! * C_{n}^{i} \) 所以维护 \( f[x][i] = \sum\limits_{u\in subtree[x],d=dist(x,u)}^{n} C_{d}^{i} \) 然后利用 \( C_{n}^{m} = C_{n-1}^{m} + C_{n-1}^{m-1} \),

【[国家集训队] Crash 的文明世界】

先写一个五十分的思路吧 首先这道题有一个弱化版 [POI2008]STA-Station 相当于\(k=1\),于是就是一个非常简单的树形\(dp\)的\(up\ \ and\ \ down\)思想 但是我们现在要求的是这个柿子了 \[\sum_{j=1}^ndis(i,j)^k\] 感觉这个东西很组合数学啊,感觉这个柿子像是天生为二项式定理准备的 我们还是考虑树形\(dp\) 在第一遍\(up\)的时候,我们设\(dp[i][k]\)表示 \[\sum_{j\in{i}}dis(i,j)^k\

Gym 101147G 第二类斯特林数

大致题意: n个孩子,k场比赛,每个孩子至少参加一场比赛,且每场比赛只能由一个孩子参加.问有多少种分配方式. 分析: k>n,就无法分配了. k<=n.把n分成k堆的方案数乘以n的阶乘.N分成k堆得方案数即第二类斯特林数 http://blog.csdn.net/acdreamers/article/details/8521134 #include <bits/stdc++.h> using namespace std; typedef long long ll; const ll

Light OJ 1236 Race 第二类斯特林数

第二类斯特林数 n 匹马 分成1 2 3... n组 每一组就是相同排名 没有先后 然后组与组之间是有顺序的 在乘以组数的阶乘 #include <cstdio> #include <cstring> using namespace std; int dp[1010][1010]; int a[1010]; int main() { a[0] = 1; dp[0][0] = 1; for(int i = 1; i <= 1000; i++) { dp[i][0] = 0; d

swjtu oj Paint Box 第二类斯特林数

http://swjtuoj.cn/problem/2382/ 题目的难点在于,用k种颜色,去染n个盒子,并且一定要用完这k种颜色,并且相邻的格子不能有相同的颜色, 打了个表发现,这个数是s(n, k) * k! s(n, k)表示求第二类斯特林数. 那么关键是怎么快速求第二类斯特林数. 这里提供一种O(k)的算法. 第二类斯特林数: #include <cstdio> #include <cstdlib> #include <cstring> #include <

hdu 2512 一卡通大冒险(第二类斯特林数)

递推思路如下,i张卡片分成j堆,那么分为两种情况:第i张卡片自成一堆或没有自成一堆. 那么自成一堆的话就是dp[i-1][j-1]种情况 不自成一堆的话就是就能在j堆种任意挑一堆放入,所以有dp[i-1][j]*j种情况 综上,如下: dp[i][j]=dp[i-1][j]*j+dp[i-1][j-1]. 关于第二类斯特林数,百度就好. 具体代码 #include <iostream> using namespace std; int dp[2005][2005]; int main() {

poj 3088 组合计数,第二类斯特林数

题意:给出n个数字[1,n],问你可以组成多少种不同的序列,其中每一个序列由几个部分组成,每个部分包含几个数字,每部分内数字无序,部分之间数字有序.每个数字最多使用一次,可以不用. 思路:枚举从n个数字中选出i个数字(组合数),再枚举将这i个数字分成j个部分(第二类斯特林数),然后乘上j的全排列. 1 #include <iostream> 2 #include <cstring> 3 #include <cstdio> 4 using namespace std; 5

LightOJ 1326 - Race(第二类斯特林数啊 )

题目链接:http://lightoj.com/volume_showproblem.php?problem=1326 百度百科:斯特林数 ACdreamer:第一类Stirling数和第二类Stirling Disky and Sooma, two of the biggest mega minds of Bangladesh went to a far country. They ate, coded and wandered around, even in their holidays.

Gym Gym 101147G 第二类斯特林数

题目链接:http://codeforces.com/gym/101147/problem/G 题意:n个人,去参加k个游戏,k个游戏必须非空,有多少种放法? 分析: 第二类斯特林数,划分好k个集合后乘以阶乘: 1 #include <bits/stdc++.h> 2 3 using namespace std; 4 5 const int maxn = 1010; 6 const long long MOD = 1000000000 + 7L; 7 long long stir[maxn][