【树链剖分】【JLOI 2014】【bzoj 3631】松鼠的新家
来源:互联网 发布:zepto.js 点击事件 编辑:程序博客网 时间:2024/04/29 19:26
3631: [JLOI2014]松鼠的新家
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 877 Solved: 421
Description
松鼠的新家是一棵树,前几天刚刚装修了新家,新家有n个房间,并且有n-1根树枝连接,每个房间都可以相互到达,且俩个房间之间的路线都是唯一的。天哪,他居然真的住在“树”上。松鼠想邀请小熊维尼前来参观,并且还指定一份参观指南,他希望维尼能够按照他的指南顺序,先去a1,再去a2,……,最后到an,去参观新家。
可是这样会导致维尼重复走很多房间,懒惰的维尼不听地推辞。可是松鼠告诉他,每走到一个房间,他就可以从房间拿一块糖果吃。维尼是个馋家伙,立马就答应了。
现在松鼠希望知道为了保证维尼有糖果吃,他需要在每一个房间各放至少多少个糖果。因为松鼠参观指南上的最后一个房间an是餐厅,餐厅里他准备了丰盛的大餐,所以当维尼在参观的最后到达餐厅时就不需要再拿糖果吃了。
Input
第一行一个整数n,表示房间个数
第二行n个整数,依次描述a1-an
接下来n-1行,每行两个整数x,y,表示标号x和y的两个房间之间有树枝相连。
Output
一共n行,第i行输出标号为i的房间至少需要放多少个糖果,才能让维尼有糖果吃。
Sample Input
51 4 5 3 21 22 42 34 5
Sample Output
12121
HINT
2<= n <=300000
题解:
树链剖分基础题(我会说我写了两个多小时才A吗)
相当于是给出一堆修改,线段树上区间+1,记得要去重就好了,最后是单点查询。
Code:
#include<iostream>#include<cstdio>#include<cstdlib>#include<cmath>#include<cstring>#include<algorithm>using namespace std;#define N 300100#define root 1,1,tot#define lch rt<<1,l,mid#define rch rt<<1|1,mid+1,rstruct Edge{ int v,next;}edge[N<<1];struct Tree{ int s,lazy;}tree[N<<2];int n,num=0,tot=0,a[N],head[N];int de[N],fa[N],sz[N],son[N],w[N],top[N];int in(){ int x=0; char ch=getchar(); while (ch<'0' || ch>'9') ch=getchar(); while (ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar(); return x;}void add(int u,int v){ edge[++num].v=v; edge[num].next=head[u]; head[u]=num;}void dfs1(int x){ sz[x]=1; for (int i=head[x]; i; i=edge[i].next){ int v=edge[i].v; if (v==fa[x]) continue; de[v]=de[x]+1; fa[v]=x; dfs1(v); sz[x]+=sz[v]; if (sz[son[x]]<sz[v]) son[x]=v; }}void dfs2(int x,int st){ w[x]=++tot; top[x]=st; if (!son[x]) return; dfs2(son[x],st); for (int i=head[x]; i; i=edge[i].next){ int v=edge[i].v; if (v==fa[x] || v==son[x]) continue; dfs2(v,v); }}void push_up(int rt){ tree[rt].s=tree[rt<<1].s+tree[rt<<1|1].s;}void push_down(int rt,int len){ tree[rt<<1].s+=(len-(len>>1))*tree[rt].lazy; tree[rt<<1|1].s+=(len>>1)*tree[rt].lazy; tree[rt<<1].lazy+=tree[rt].lazy; tree[rt<<1|1].lazy+=tree[rt].lazy; tree[rt].lazy=0;}void change(int rt,int l,int r,int ll,int rr,int k){ if (ll<=l && r<=rr){ tree[rt].s+=(r-l+1)*k; tree[rt].lazy+=k; return; } if (tree[rt].lazy!=0) push_down(rt,r-l+1); int mid=(l+r)>>1; if (ll<=mid) change(lch,ll,rr,k); if (rr>mid) change(rch,ll,rr,k); push_up(rt);}int query(int rt,int l,int r,int x){ if (l==r) return tree[rt].s; if (tree[rt].lazy!=0) push_down(rt,r-l+1); int mid=(l+r)>>1; if (x<=mid) return query(lch,x); return query(rch,x);}void work(int x,int y){ int yy=y; if (de[top[x]]<de[top[y]]) swap(x,y); while (top[x]!=top[y]){ if (de[top[x]]<de[top[y]]) swap(x,y); change(root,w[top[x]],w[x],1); x=fa[top[x]]; } if (de[x]<de[y]) swap(x,y); change(root,w[y],w[x],1); change(root,w[yy],w[yy],-1);}int main(){ n=in(); for (int i=1; i<=n; i++) a[i]=in(); for (int i=1; i<n; i++){ int u=in(),v=in(); add(u,v),add(v,u); } dfs1(1),dfs2(1,1); for (int i=1; i<n; i++) work(a[i],a[i+1]); for (int i=1; i<=n; i++) printf("%d\n",query(root,w[i])); return 0;}
1 0
- BZOJ 3631 JLOI 2014 松鼠的新家 LCA 树链剖分
- 【树链剖分】【JLOI 2014】【bzoj 3631】松鼠的新家
- 【BZOJ 3631】【JLOI 2014】松鼠的新家【树链剖分】
- bzoj 3631: [JLOI2014]松鼠的新家
- 【BZOJ 3631】 [JLOI2014]松鼠的新家
- bzoj 3631 松鼠的新家
- bzoj 3631: [JLOI2014]松鼠的新家
- bzoj 3631 松鼠的新家
- bzoj 3631 [JLOI2014]松鼠的新家
- BZOJ 3631 松鼠的新家
- BZOJ 3631 [JLOI2014]松鼠的新家
- bzoj 3631 [JLOI2014]松鼠的新家
- BZOJ 3631 JLOI2014 松鼠的新家 树链剖分/LCA
- bzoj 3631: [JLOI2014]松鼠的新家 树链剖分
- 【bzoj 3631】[JLOI2014]松鼠的新家(树链剖分)
- Bzoj 3631: [JLOI2014]松鼠的新家(树链剖分+线段树)
- BZOJ 3631 [JLOI2014]松鼠的新家 树链剖分
- BZOJ 3631 [JLOI2014]松鼠的新家 tarjanlca
- java WEB应用程序启动时是如何加载AXIS server-config.wsdd 的
- 摄影构图常用技巧
- 转载好文
- Android利用自定义View实现水波纹效果
- pom.xml文件详细的配置方法
- 【树链剖分】【JLOI 2014】【bzoj 3631】松鼠的新家
- U3D学习笔记(6)
- 设计模式笔记(九)--状态模式、代理模式
- VPN远程访问概述
- EF框架实体类映射成数据库字段的办法,未能找到类型或命名空间名称“NotMapped”(是否缺少 using 指令或程序集引用?)
- Java代码调用数据库带输出参数的存储过程
- IOS中NSUserDefaults的用法(轻量级本地数据存储)
- tomcat的Context配置
- server-config.wsdd配置一例