先写一个五十分的思路吧

首先这道题有一个弱化版

[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
\]

\(j\in{i}\)表示\(j\)在\(i\)子树内部

于是我们考虑一下化这个柿子

到达\(i\)肯定要先达到\(i\)的一个儿子,于是就有

\[dp[i][k]=\sum_{fa[j]=i}\sum_{t\in{j}}(dis(t,j)+1)^k
\]

我们用二项式定理来将这个柿子展开

\[dp[i][k]=\sum_{fa[j]=i}\sum_{t\in{j}}\sum_{r=0}^kC_{k}^{r}*dis(t,j)^r
\]

后面两个\(\sum\)换一下位置

\[dp[i][k]=\sum_{fa[j]=i}\sum_{r=0}^kC_{k}^{r}*\sum_{t\in{j}}dis(t,j)^r
\]

之后就会惊奇的发现\(\sum_{t\in{j}}dis(t,j)^r\)就是\(dp[j][r]\),于是现在就有了

\[dp[i][k]=\sum_{fa[j]=i}\sum_{r=0}^kC_{k}^{r}*dp[j][r]
\]

这就是\(up\)的转移方程式,\(down\)的方程式也很好推

\(down\)的时候\(dp[i][k]\)表示的不仅局限于\(i\)的子树内部了,而是整棵树了

到达\(i\)首先要到达\(fa[i]\),于是就有

\[dp[i][k]+=\sum_{j\notin{i},j\in{fa[i]}}(dis(fa[i],j)+1)^k
\]

\[dp[i][k]+=\sum_{j\notin{i},j\in{fa[i]}}\sum_{r=0}^kC_{k}^r*dis(fa[i],j)^r
\]

我们的要求不就是\(j\)不能来自于\(i\)内部吗,于是我们大力容斥就好了

我们把来自于\(i\)子树内部的答案减掉,于是就有了一个可以更新的柿子了

现在就有了一个我们就可以转移了,复杂度大概是\(O(nk^2)\)

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#define re register
#define maxn 50005
#define int long long
const int mod=10007;
inline int read()
{
char c=getchar();
int x=0;
while(c<'0'||c>'9') c=getchar();
while(c>='0'&&c<='9')
x=(x<<3)+(x<<1)+c-48,c=getchar();
return x;
}
struct E
{
int v,nxt;
}e[maxn<<1];
int head[maxn],deep[maxn];
int n,m,num;
long long dp[maxn][151];
long long now[151];
long long c[151][151];
void dfs(int x)
{
dp[x][0]=1;
for(re int i=head[x];i;i=e[i].nxt)
if(!deep[e[i].v])
{
deep[e[i].v]=deep[x]+1;
dfs(e[i].v);
dp[x][0]+=dp[e[i].v][0];
for(re int k=1;k<=m;k++)
for(re int r=0;r<=k;r++)
dp[x][k]=(dp[x][k]+c[k][r]*dp[e[i].v][r])%mod;
}
}
void redfs(int x)
{
for(re int i=head[x];i;i=e[i].nxt)
if(deep[e[i].v]>deep[x])
{
memset(now,0,sizeof(now));
for(re int k=0;k<=m;k++)
{
now[k]=dp[x][k];
for(re int r=0;r<=k;r++)
now[k]=(now[k]-dp[e[i].v][r]*c[k][r]+mod)%mod;
}//先容斥,不能来自于e[i].v子树内部
for(re int k=0;k<=m;k++)
for(re int r=0;r<=k;r++)
dp[e[i].v][k]=(dp[e[i].v][k]+c[k][r]*now[r])%mod;//用容斥之后的答案来更新
redfs(e[i].v);
}
}
inline void add_edge(int x,int y)
{
e[++num].v=y;
e[num].nxt=head[x];
head[x]=num;
}
signed main()
{
n=read(),m=read();
int x,y;
for(re int i=1;i<n;i++)
x=read(),y=read(),add_edge(x,y),add_edge(y,x);
c[0][0]=1;
for(re int i=1;i<=m;i++) c[i][0]=c[i][i]=1;
for(re int i=1;i<=m;i++)
for(re int j=1;j<i;j++) c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
deep[1]=1;
dfs(1);
redfs(1);
for(re int i=1;i<=n;i++)
printf("%lld\n",(dp[i][m]%mod+mod)%mod);
return 0;
}

这是一个要T的复杂度,但是已经有了\(50\)分的好成绩

点分治据说可以做到\(O(nklogk)\),但是不会

我们继续组合数做吧

我们发现求\(x^k\)可以理解为把\(k\)个物品放到\(x\)个互不相同的盒子里,允许有盒子空着不放的方案数

于是我们可以写成\(x^k=\sum_{i=1}^kS(k,i)*C_{x}^i*i!\)

其中\(S(k,i)\)是第二类斯特林数,表示的是将\(k\)个球分到\(i\)个盒子里,这\(i\)个盒子没有差别,而且没有盒子是空的的方案数

\(C_{x}^i*i!\)其实就是排列数了,就相当于我们给\(i\)个盒子强行制造了差别

于是这个柿子可以理解为\(i\)枚举的是当前有几个盒子是有球的,之后通过加法原理合并了答案

其实我一开始觉得这里的\(\sum\)的上标应该写\(x\),好像也只有写\(x\)才满足组合的意义,之后发现自己非常naive

  1. 当\(k>x\)的时候显然是没有什么问题的了,因为\(\binom{x}{i}\)在\(x>i\)的时候取0,于是没有什么影响

  2. 当\(k<x\)的时候,如果上标取到比\(k\)大的数了,那么也会导致\(S(k,i)\)变成\(0\),于是写成\(k\)就可以了,在数值上没有什么影响

之后我们继续化柿子

\[ans[t]=\sum_{j=1}^{n}dis(t,j)^k
\]

\[=\sum_{j=1}^{n}\sum_{i=1}^kS(k,i)*\binom{dis(t,j)}{i}*i!
\]

\[=\sum_{i=1}^{k}S(k,i)*i!*\sum_{j=1}^n\binom{dis(t,j)}{i}
\]

那么我们现在只需要求出\(\sum_{j=1}^n\binom{dis(t,j)}{i}\)就好了

我们都知道组合数有一个非常好的转移的方式就是\(\binom{n}{m}=\binom{n-1}{m-1}+\binom{n-1}{m}\)

于是

\[\sum_{j=1}^n\binom{dis(t,j)}{i}=\sum_{j=1}^n\binom{dis(t,j)-1}{i-1}+\sum_{j=1}^n\binom{dis(t,j)-1}{i}
\]

现在是不是又可以用树形dp来转移了,因为到达\(t\)这个点还是要先到达\(t\)的儿子或者是父亲

于是我们设\(dp[x][k]=\sum_{j\in{x}}\binom{dis(x,i)}{k}\)

于是在\(up\)里的方程式就是

\[dp[x][k]=\sum_{fa[j]=i}dp[j][k]+dp[j][k-1]
\]

\(down\)里我们还是要先容斥一下,方程和\(up\)类似

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#define re register
#define maxn 50005
#define LL long long
const LL mod=10007;
inline int read()
{
char c=getchar();
int x=0;
while(c<'0'||c>'9') c=getchar();
while(c>='0'&&c<='9')
x=(x<<3)+(x<<1)+c-48,c=getchar();
return x;
}
struct node
{
int v,nxt;
}e[maxn<<1];
int head[maxn],deep[maxn];
LL dp[maxn][151],fac[151],s[151][151];
LL now[151];
int n,m,num;
inline void add_edge(int x,int y)
{
e[++num].v=y;
e[num].nxt=head[x];
head[x]=num;
}
void dfs(int x)
{
dp[x][0]=1;
for(re int i=head[x];i;i=e[i].nxt)
if(!deep[e[i].v])
{
deep[e[i].v]=deep[x]+1;
dfs(e[i].v);
dp[x][0]+=dp[e[i].v][0];
for(re int j=1;j<=m;j++)
dp[x][j]=(dp[x][j]+dp[e[i].v][j-1]+dp[e[i].v][j])%mod;
}
}
void redfs(int x)
{
for(re int i=head[x];i;i=e[i].nxt)
if(deep[e[i].v]>deep[x])
{
memset(now,0,sizeof(now));
now[0]=dp[x][0]-dp[e[i].v][0];
for(re int j=1;j<=m;j++)
now[j]=(dp[x][j]-dp[e[i].v][j]-dp[e[i].v][j-1]+mod)%mod;
//依旧是先容斥一遍
dp[e[i].v][0]+=now[0];
for(re int j=1;j<=m;j++)
dp[e[i].v][j]=(dp[e[i].v][j]+now[j]+now[j-1])%mod;
redfs(e[i].v);
}
}
int main()
{
n=read(),m=read();
int x,y;
for(re int i=1;i<n;i++)
x=read(),y=read(),add_edge(x,y),add_edge(y,x);
s[0][0]=1;
for(re int i=1;i<=m;i++)
s[i][1]=s[i][i]=1;
for(re int i=1;i<=m;i++)
for(re int j=1;j<i;j++)
s[i][j]=(s[i-1][j-1]+s[i-1][j]*j)%mod;//预处理第二类斯特林数
fac[0]=1;
for(re int i=1;i<=m;i++)
fac[i]=(fac[i-1]*i)%mod;//预处理阶乘
deep[1]=1;
dfs(1);
redfs(1);
for(re int i=1;i<=n;i++)
for(re int j=1;j<=m;j++)
dp[i][j]=(dp[i][j]%mod+mod)%mod;
for(re int i=1;i<=n;i++)
{
LL ans=0;
for(re int j=1;j<=m;j++)
ans=(ans+s[m][j]*fac[j]%mod*dp[i][j])%mod;
printf("%lld\n",ans);//统计答案
}
return 0;
}

【[国家集训队] Crash 的文明世界】的更多相关文章

  1. [国家集训队] Crash 的文明世界(第二类斯特林数)

    题目 [国家集训队] Crash 的文明世界 前置 斯特林数\(\Longrightarrow\)斯特林数及反演总结 做法 \[\begin{aligned} ans_x&=\sum\limi ...

  2. [国家集训队] Crash的文明世界

    Description 给定一棵 \(n\) 个点的树,对于每个点 \(i\) 求 \(S(i)=\sum\limits_{j=1}^n \operatorname{dist(i,j)}^k\) .\ ...

  3. [国家集训队] Crash 的文明世界

    不错的树形$ DP$的题 可为什么我自带大常数啊$ cry$ 链接:here 题意:给定一棵$ n$个节点的树,边权为$ 1$,对于每个点$ x$求$ \sum\limits_{i=1}^n dist ...

  4. 洛谷P4827 [国家集训队] Crash 的文明世界 [斯特林数,组合数,DP]

    传送门 思路 又见到这个\(k\)次方啦!按照套路,我们将它搞成斯特林数: \[ ans_x=\sum_{i=0}^k i!S(k,i)\sum_y {dis(x,y) \choose i} \] 前 ...

  5. P4827 [国家集训队] Crash 的文明世界

    传送门:洛谷 题目大意:设$$S(i)=\sum_{j=1}^ndis(i,j)^k$$,求$S(1),S(2),\ldots,S(n)$. 数据范围:$n\leq 50000,k\leq 150$ ...

  6. 解题:国家集训队 Crash 的文明世界

    题面 这种套着高次幂的统计问题一般都要用到第二类斯特林数和自然数幂的关系:$a^k=\sum\limits_{i=0}^{k}S_k^iC_a^i*i!$ 那么对于每个点$x$有: $ans_x=\s ...

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

    传送门 对于点\(u\),所求为\[\sum_{i=1}^ndis(i,u)^k\] 把后面那堆东西化成第二类斯特林数,有\[\sum_{i=1}^n\sum_{j=0}^kS(k,j)\times ...

  8. 国家集训队 Crash 的文明世界(第二类斯特林数+换根dp)

    题意 ​ 题目链接:https://www.luogu.org/problem/P4827 ​ 给定一棵 \(n\) 个节点的树和一个常数 \(k\) ,对于树上的每一个节点 \(i\) ,求出 \( ...

  9. 洛谷 P4827 [国家集训队] Crash 的文明世界

    题目描述 ​ 给你一棵 n 个点的树,对于树上的每个节点 i,求 \(\sum_{j=1}^ndis(i,j)^k\).其中 \(dis(i,j)\) 为两点在树上的距离. 输入格式 ​ 第一行两个整 ...

随机推荐

  1. Pnel控件

    分组类控件 面板控件(Panel) 分组框控件(GroupBox) 选项卡控件(TabControl)等控件   Panel 控件是由System.Windows.Forms.Panel类提供的,主要 ...

  2. sql prompt 不能用

    问题描述: 安装成功后,打开sql server 工具栏不显示菜单,并弹出提示错误信息: SQL Prompt has been disabled due to an error with the r ...

  3. 961 -尺寸2N阵列中的N重复元素

    在一个A大小的数组中2N,有N+1独特的元素,这些元素中的一个重复N次. 返回重复N次的元素. 例1: 输入:[1,2,3,3] 输出:3 例2: 输入:[2,1,2,5,3,2] 输出:2 例3: ...

  4. DB2 Metadata

    http://www.devart.com/dotconnect/db2/docs/MetaData.html Instead of specifying the metadata collectio ...

  5. luogu P3065 first——trie树相关

    题目描述 Bessie has been playing with strings again. She found that by changing the order of the alphabe ...

  6. 远景平台开发者上线,专业API免费使用

    远景平台开发者上线,欢迎大伙围观使用. 在开发者中心你可以做什么? 1.管理你的应用,通过APPKEY获取在线API.使用云中的数据和地图. 2.学习API的使用,包含API参考和部分例子(目前例子很 ...

  7. IEC62304-2006解读

    IEC62304强调医疗软件在明确和满足其预期用途的前提下,不能引发不可接受的风险 62304提供一个医疗软件开发的框架,并指出框架下每个过程的要求,62304将过程分解为若干活动,活动分解为若干任务 ...

  8. 关于asp.net MVC3 ----@Html.Partial,@Html.Action,@Html.RenderPartial,@Html.RenderAction

    1.带有Render的方法返回值是void,在方法内部进行输出:不带的返回值类型为MvcHtmlString,所以只能这样使用:@Html.Partial 对应 @{Html.RenderPartia ...

  9. Java 8 Date-Time API概览

    更新时间:2018-04-19 根据网上资料整理 java 8增加了新的Date-Time API (JSR 310),增强对日期与时间的处理.它在很大程度上受到Joda-Time的影响.之前写过一篇 ...

  10. 五、python小功能记录——打包程序

    使用pyinstaller打包Python程序 安装工具 :pip3 install pyinstaller 在Python程序文件夹上(不点进去)按住shift并且右键,在弹出的选项中点击" ...