BZOJ

洛谷


挺套路但并不难的一道题

\(Description\)

给定一棵\(n\)个点的树和\(K\),边权为\(1\)。对于每个点\(x\),求\(S(x)=\sum_{i=1}^ndis(x,i)^K\)。

\(n\leq50000,\ k\leq150\)。

\(Solution\)

和其它求\(x^k\)的题一样,依旧用第二类斯特林数展开。(二项式定理依旧可以得到部分分,依旧不想看=-=)

\[\begin{aligned}S(x)&=\sum_{i=1}^ndis(x,i)^K\\&=\sum_{i=1}^n\sum_{k=0}^{dis(x,i)}S(K,k)\cdot k!\cdot \binom{dis(x,i)}{k}\\&=\sum_{i=1}^n\sum_{k=0}^KS(K,k)\cdot k!\cdot \binom{dis(x,i)}{k}\\&=\sum_{k=0}^KS(K,k)\cdot k!\cdot\sum_{i=1}^n\binom{dis(x,i)}{k}\end{aligned}
\]

考虑这个\(\sum_{i=1}^n\binom{dis(x,i)}{k}\)怎么算。用阶乘公式拆还是一样没法算,尝试用这个公式拆:\(\binom nm=\binom{n-1}m\times\binom{n-1}{m-1}\):

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

记\(f[x][k]=\sum_{i=1}^n\binom{dis(x,i)}{k}\),那么可以由\(x\)的相邻点\(v\)的\(f[v][k]+f[v][k-1]\)转移来(\(x\)和\(v\)与其它点的\(dis\)正好差\(1\))。

具体就是两遍树形DP,第一次自底向上求出子树内的点到\(x\)的\(dis\)的贡献,即\(f[x][i]=\sum_{v\in son[x]}f[v][i]+f[v][i-1]\);第二次从上往下更新子树外的点到\(v=son[x]\)的\(dis\)的贡献,记为\(g[v][i]=g[x][i]+g[x][i-1]+(f[x][i]-f[v][i]-f[v][i-1])+(f[x][i-1]-f[v][i-1]-f[v][i-2])\)。

然后统计一下就OK了。复杂度\(O(nk+k^2)\)。

明明取模优化的不少啊,怎么就这么慢呢=-=


//33844kb	4868ms
#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
#define mod 10007
#define Mod(x) x>=mod&&(x-=mod)
#define Add(x,v) (x+=v)>=mod&&(x-=mod)
typedef long long LL;
const int N=50003,M=153; int K,Enum,H[N],nxt[N<<1],to[N<<1],f[N][M]; inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-48,c=gc());
return now;
}
inline void AE(int u,int v)
{
to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum;
}
void DFS1(int x,int fa)
{
f[x][0]=1;//这个初始化...C(dis(x,x),0)=1?有点小懵逼=-=
int K=::K;
for(int i=H[x],v; i; i=nxt[i])
if((v=to[i])!=fa)
{
DFS1(v,x), f[x][0]+=f[v][0];
for(int j=1; j<=K; ++j) f[x][j]+=f[v][j]+f[v][j-1];
}
for(int i=0; i<=K; ++i) f[x][i]%=mod;
}
void DFS2(int x,int fa)
{
static int g[N];
int K=::K;
for(int i=H[x],v; i; i=nxt[i])
if((v=to[i])!=fa)
{
g[0]=f[x][0]+mod-f[v][0];
for(int j=1; j<=K; ++j) g[j]=f[x][j]+mod-f[v][j]+mod-f[v][j-1];//g[j] = g[x][j]+f[x][j]-f[v][j]-f[v][j-1]
f[v][0]=(f[v][0]+g[0])%mod;
for(int j=1; j<=K; ++j) f[v][j]=(f[v][j]+g[j]+g[j-1])%mod;
DFS2(v,x);
}
} int main()
{
static int S[M][M];
const int n=read(),K=read(); ::K=K;
// for(int i=1; i<n; ++i) AE(read(),read());
for(int L=read(),now=read(),A=read(),B=read(),Q=read(),i=1; i<n; ++i)
now=(now*A+B)%Q, AE(i-now%(i<L?i:L),i+1); DFS1(1,1), DFS2(1,1), S[0][0]=1;
for(int i=1; i<=K; ++i)
for(int j=1; j<=i; ++j) S[i][j]=(S[i-1][j-1]+S[i-1][j]*j)%mod;
for(int x=1; x<=n; ++x)
{
LL ans=0;
for(int i=0,fac=1; i<=K; ++i) ans+=1ll*S[K][i]*fac*f[x][i]%mod, fac=fac*(i+1)%mod;
printf("%d\n",(int)(ans%mod));
} return 0;
}

BZOJ.2159.Crash的文明世界(斯特林数 树形DP)的更多相关文章

  1. 【BZOJ2159】Crash的文明世界 斯特林数+树形dp

    Description Crash 小朋友最近迷上了一款游戏--文明5(Civilization V).在这个游戏中,玩家可以建立和发展自己的国家,通过外交和别的国家交流,或是通过战争征服别的国家.现 ...

  2. [BZOJ2159]Crash的文明世界(斯特林数+树形DP)

    题意:给定一棵树,求$S(i)=\sum_{j=1}^{n}dist(i,j)^k$.题解:根据斯特林数反演得到:$n^m=\sum_{i=0}^{n}C(n,i)\times i!\times S( ...

  3. BZOJ 2159: Crash 的文明世界(树形dp+第二类斯特林数+组合数)

    题意 给定一棵 \(n\) 个点的树和一个常数 \(k\) , 对于每个 \(i\) , 求 \[\displaystyle S(i) = \sum _{j=1} ^ {n} \mathrm{dist ...

  4. bzoj 2159 Crash 的文明世界 && hdu 4625 JZPTREE ——第二类斯特林数+树形DP

    题目:https://www.lydsy.com/JudgeOnline/problem.php?id=2159 学习材料:https://blog.csdn.net/litble/article/d ...

  5. BZOJ 2159: Crash 的文明世界(组合数学+第二类斯特林数+树形dp)

    传送门 解题思路 比较有意思的一道数学题.首先\(n*k^2\)的做法比较好想,就是维护一个\(x^i\)这种东西,然后转移的时候用二项式定理拆开转移.然后有一个比较有意思的结论就是把求\(x^i\) ...

  6. bzoj 2159 Crash 的文明世界 & hdu 4625 JZPTREE —— 第二类斯特林数+树形DP

    题目:https://www.lydsy.com/JudgeOnline/problem.php?id=2159 使用公式:\( n^{k} = \sum\limits_{i=0}^{k} S(k,i ...

  7. bzoj 2159: Crash 的文明世界

    Time Limit: 10 Sec  Memory Limit: 259 MB Submit: 480  Solved: 234[Submit][Status][Discuss] Descripti ...

  8. [bzoj 2159]Crash的文明世界

    今天看到一个鬼题 心情好的时候写 [题意]求树上所有点对距离的k次方和,所有边权为1 大爷方的题解:http://tonyfang.is-programmer.com/posts/204972.htm ...

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

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

随机推荐

  1. Datagrip导入导出为一个sql文件详细说明 (mysql)

    Datagrip导入导出为一个sql文件详细说明 导出: (我的无法截图, 借用一张) 导入: (我的无法截图, 再借用一张) 完

  2. linux下mysql源码安装

    参考链接:http://blog.csdn.net/zqtsx/article/details/9378703 下载mysql安装包, 不会下载点这里 地址:ftp://mirror.switch.c ...

  3. FCN 项目部分代码学习

    下面代码由搭档注释,保存下来用作参考. github项目地址:https://github.com/shekkizh/FCN.tensorflowfrom __future__ import prin ...

  4. python 装饰器(1)

    # 装饰器,对一个函数打扮 # def foo():# print("foo")# foo# foo()# def test1():# print('-----1-----')# ...

  5. SQL Server等待

    等待大概分为3类:资源等待.队列等待.外部等待 过滤掉系统相关的等待类型的语句.(查看常用的等待信息) SELECT wait_type , signal_wait_time_ms , wait_ti ...

  6. IDEAL 中配置Tomcat的内存值

    -server -XX:PermSize=2048M -XX:MaxPermSize=4096m

  7. Git基础(二) 文件的生命周期

    使用Git时,文件的生命周期如下:

  8. DBEntityEntry类

    DBEntityEntry是一个重要的类,可用于检索有关实体的各种信息.您可以使用DBContext的Entry方法获取特定实体的DBEntityEntry实例. DBEntityEntry允许您访问 ...

  9. babelrc

    .babelrc文件 // 简单版 { "presets": ["es2015", "stage-2"], // 使用 es2015 npm ...

  10. PHP Fatal error: SOAP-ERROR: Parsing WSDL: Couldn't load from 'http://xxxx.wsdl'

    libxml_disable_entity_loader(false); $client = new \SoapClient($wsdl); 完美解决办法加上 php的soap扩展是否安装  open ...