bzoj3631[JLOI2014 松鼠的新家 倍增lca+差分
2017-08-22 16:38
302 查看
裸的树上差分+倍增lca
每次从起点到终点左闭右开,这就有一个小技巧,要找到右端点向左端点走的第一步,然后差分就好了
每次从起点到终点左闭右开,这就有一个小技巧,要找到右端点向左端点走的第一步,然后差分就好了
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #include<cmath> #define N 300005 using namespace std; int fa [20],dep ,f ,g ,n,l ; int e=1,head ; struct edge{ int u,v,next; }ed[2*N]; void add(int u,int v){ ed[e].u=u;ed[e].v=v; ed[e].next=head[u]; head[u]=e++; } void dfs(int x){ for(int i=1;(1<<i)<=dep[x];i++) fa[x][i]=fa[fa[x][i-1]][i-1]; for(int i=head[x];i;i=ed[i].next){ int v=ed[i].v; if(v==fa[x][0])continue; fa[v][0]=x;dep[v]=dep[x]+1; dfs(v); } } int lca(int x,int y){ if(dep[x]<dep[y])swap(x,y); for(int i=19;i>=0;i--) if(dep[fa[x][i]]>=dep[y]) x=fa[x][i]; if(x==y)return x; for(int i=19;i>=0;i--) if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i]; return fa[x][0]; } int find(int x,int y){ int d=dep[x]-y; for(int i=19;~i;i--) if(dep[fa[x][i]]>=d) x=fa[x][i]; return x; } void update(int x,int y){ f[x]++;f[y]++; int z=lca(x,y); f[z]--;f[fa[z][0]]--; } void work(int x,int y){ if(dep[x]<=dep[y])update(x,fa[y][0]); else{ int d=dep[x]-dep[y]; int z=find(x,d-1); if(fa[z][0]==y)update(x,z); else update(x,fa[y][0]); } } void dfs1(int x){ g[x]=f[x]; for(int i=head[x];i;i=ed[i].next){ if(ed[i].v==fa[x][0])continue; dfs1(ed[i].v); g[x]+=g[ed[i].v]; } } int main(){ scanf("%d",&n); int u,v; for(int i=1;i<=n;i++)scanf("%d",&l[i]); for(int i=1;i<n;i++){ scanf("%d%d",&u,&v); add(u,v);add(v,u); } dep[1]=1; dfs(1); for(int i=1;i<n;i++) work(l[i],l[i+1]); dfs1(1); for(int j=1;j<=n;j++) printf("%d\n",g[j]); return 0; }
相关文章推荐
- bzoj3631[JLOI2014 松鼠的新家 倍增lca+差分
- [JLOI2014][BZOJ3631] 松鼠的新家|树上倍增LCA|差分
- 【bzoj3631】[JLOI2014]松鼠的新家 LCA+差分数组
- BZOJ3631 [JLOI2014] 松鼠的新家
- [BZOJ3631][洛谷P3258][JLOI2014]松鼠的新家
- BZOJ 3631: [JLOI2014]松鼠的新家( 树链剖分 )
- BZOJ 3631: [JLOI2014]松鼠的新家【差分】【LCA】
- bzoj 3631: [JLOI2014]松鼠的新家
- BZOJ 3631 [JLOI2014]松鼠的新家 | 树上差分
- 【BZOJ 3631】 [JLOI2014]松鼠的新家
- BZOJ 3631 [JLOI2014]松鼠的新家==树剖
- [bzoj3631][JLOI2014]松鼠的新家 树上差分
- BZOJ 3631 [JLOI2014]松鼠的新家 tarjanlca
- BZOJ 3631: [JLOI2014]松鼠的新家 树上差分
- bzoj 3631: [JLOI2014]松鼠的新家
- 洛谷 P3258 BZOJ 3631 [JLOI2014]松鼠的新家
- [BZOJ3631][JLOI2014]松鼠的新家(链剖)
- BZOJ 3631 JLOI2014 松鼠的新家 树链剖分/LCA
- BZOJ 3631 [JLOI2014]松鼠的新家
- 【bzoj 3631】[JLOI2014]松鼠的新家(树链剖分)