博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
BZOJ.4543.[POI2014]Hotel加强版(长链剖分 树形DP)
阅读量:5065 次
发布时间:2019-06-12

本文共 2069 字,大约阅读时间需要 6 分钟。

弱化版:。

\(f[x][i]\)表示\(x\)的子树中深度为\(i\)的点的个数,\(g[x][i]\)表示\(x\)子树中,满足\(u,v\)\(LCA(u,v)\)的距离都是\(d\),且到\(x\)的距离为\(d-i\)的点对\((u,v)\)个数。(就是不以\(x\)作为三个点的中心位置,那样就没法算了)

1143196-20181126150148166-490422851.png

那么就可以由\(g[x][i]\)与另一棵子树的\(f[y][i-1]\),以及\(g[x][0]\)更新答案。

\(f\)的转移简单。\(g[x][i]\)要么是从某个子树中得到(\(g[x][i]+=g[v][i+1]\)),要么是以x作为LCA从某两个子树中得到(\(g[x][i]+=f[x][i]*f[v][i-1]\))。

直接这样复杂度\(O(n^2)\)

DP数组的下标都是深度,所以可以用长链剖分。继承重儿子时\(f\)需要右移,\(g\)需要左移,还是用指针方便些吧(\(g\)的前后都需要留空间)。

复杂度\(O(n)\)

为啥洛谷上跑的不快啊(30ms)==没道理啊。

//7788kb    376ms#include 
#include
#include
//#define gc() getchar()#define MAXIN 300000#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)typedef long long LL;const int N=1e5+5;int Enum,H[N],nxt[N<<1],to[N<<1],dep[N],mxd[N],son[N],pos[N],*f[N],Fp[N],*fp=Fp;LL Ans,*g[N],Gp[N<<1],*gp=Gp;char IN[MAXIN],*SS=IN,*TT=IN;inline int read(){ int now=0;register char c=gc(); for(;!isdigit(c);c=gc()); for(;isdigit(c);now=now*10+c-'0',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;}inline void Allot(int x){ f[x]=fp, fp+=mxd[x], gp+=mxd[x], g[x]=gp, gp+=mxd[x];}void DFS1(int x,int fa){ int mx=0; for(int i=H[x],v; i; i=nxt[i]) if((v=to[i])!=fa) dep[v]=dep[x]+1, DFS1(v,x), mxd[v]>mx&&(mx=mxd[v],son[x]=v); mxd[x]=mx+1;// for(int i=H[x],v; i; i=nxt[i])// if((v=to[i])!=fa) dep[v]=dep[x]+1, DFS1(v,x), mxd[v]>mxd[son[x]]&&(son[x]=v);// mxd[x]=mxd[son[x]]+1;}void DFS2(int x,int fa){ if(son[x]) f[son[x]]=f[x]+1, g[son[x]]=g[x]-1, DFS2(son[x],x); f[x][0]=1, Ans+=g[x][0];//重儿子子树与x的贡献 for(int i=H[x],v; i; i=nxt[i]) if((v=to[i])!=fa&&v!=son[x]) { Allot(v), DFS2(v,x); int l=mxd[v]; for(int j=1; j<=l; ++j) Ans+=g[x][j]*f[v][j-1]+g[v][j]*f[x][j-1]; for(int j=1; j<=l; ++j) g[x][j]+=g[v][j+1]+1ll*f[x][j]*f[v][j-1],f[x][j]+=f[v][j-1];//g[x][0]转移来也没啥用啊 }}int main(){ int n=read(); for(int i=1; i

转载于:https://www.cnblogs.com/SovietPower/p/10020366.html

你可能感兴趣的文章
2011年12月09日
查看>>
[ZJOI2007]棋盘制作 【最大同色矩形】
查看>>
合并单元格
查看>>
IOS-图片操作集合
查看>>
Android bitmap图片处理
查看>>
Android应用程序进程启动过程的源代码分析
查看>>
Redis学习手册(Key操作命令)
查看>>
模板统计LA 4670 Dominating Patterns
查看>>
泛型第23条:请不要在新代码中使用原生态类型
查看>>
从下周开始就要采用网上记录值班日志了
查看>>
团队项目开发客户端——登录子系统的设计
查看>>
【AppScan心得】IBM Rational AppScan 无法记录登录序列
查看>>
[翻译] USING GIT IN XCODE [4] 在XCODE中使用GIT[4]
查看>>
简化通知中心的使用
查看>>
IO—》Properties类&序列化流与反序列化流
查看>>
html 简介
查看>>
python使用上下文对代码片段进行计时,非装饰器
查看>>
【bzoj5099】[POI2018]Pionek 双指针法
查看>>
session如何保存在专门的StateServer服务器中
查看>>
react展示数据
查看>>