P3258 [JLOI2014]松鼠的新家
题目描述
松鼠的新家是一棵树，前几天刚刚装修了新家，新家有n个房间，并且有n-1根树枝连接，每个房间都可以相互到达，且俩个房间之间的路线都是唯一的。天哪，他居然真的住在”树“上。
松鼠想邀请小熊维尼前来参观，并且还指定一份参观指南，他希望维尼能够按照他的指南顺序，先去a1，再去a2，......，最后到an，去参观新家。可是这样会导致维尼重复走很多房间，懒惰的维尼不停地推辞。可是松鼠告诉他，每走到一个房间，他就可以从房间拿一块糖果吃。
维尼是个馋家伙，立马就答应了。现在松鼠希望知道为了保证维尼有糖果吃，他需要在每一个房间各放至少多少个糖果。
因为松鼠参观指南上的最后一个房间an是餐厅，餐厅里他准备了丰盛的大餐，所以当维尼在参观的最后到达餐厅时就不需要再拿糖果吃了。
输入输出格式
输入格式：
第一行一个整数n，表示房间个数第二行n个整数，依次描述a1-an
接下来n-1行，每行两个整数x，y，表示标号x和y的两个房间之间有树枝相连。
输出格式：
一共n行，第i行输出标号为i的房间至少需要放多少个糖果，才能让维尼有糖果吃。
树链剖分+差分数组，很裸的一个题，注意细节即可
#include <cstdio>
int max(int x,int y){return x>y?x:y;}
int min(int x,int y){return x<y?x:y;}
const int N=300010;
int to[N<<1],next[N<<1],cnt,head[N];
void add(int u,int v)
{
to[++cnt]=v;next[cnt]=head[u];head[u]=cnt;
}
int ws[N],siz[N],f[N],dfn[N],top[N],dep[N],ha[N],d[N],time=0;
int ans[N],n,a[N];
void dfs1(int now)
{
for(int i=head[now];i;i=next[i])
{
int v=to[i];
if(v!=f[now])
{
f[v]=now;
dep[v]=dep[now]+1;
dfs1(v);
siz[now]+=siz[v];
if(siz[ws[now]]<siz[v]) ws[now]=v;
}
}
siz[now]++;
}
void dfs2(int now,int anc)
{
dfn[now]=++time;
top[now]=anc;
ha[time]=now;
if(!ws[now]) return;
dfs2(ws[now],anc);
for(int i=head[now];i;i=next[i])
{
int v=to[i];
if(!dfn[v])
dfs2(v,v);
}
}
void t_add(int x,int y)
{
while(top[x]!=top[y])
{
if(dep[top[x]]>dep[top[y]])
{
d[dfn[top[x]]]++;
d[dfn[x]+1]--;
x=f[top[x]];
}
else
{
d[dfn[top[y]]]++;
d[dfn[y]+1]--;
y=f[top[y]];
}
}
d[min(dfn[x],dfn[y])]++;
d[max(dfn[x],dfn[y])+1]--;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",a+i);
int u,v;
for(int i=1;i<n;i++)
{
scanf("%d%d",&u,&v);
add(u,v),add(v,u);
}
dfs1(1);
dfs2(1,1);
for(int i=1;i<n;i++)
t_add(a[i],a[i+1]);
for(int i=1;i<=n;i++)
{
d[i]+=d[i-1];
ans[ha[i]]=d[i];
if(ha[i]!=a[1]) ans[ha[i]]--;
}
for(int i=1;i<=n;i++)
printf("%d
",ans[i]);
return 0;
}
2018.6.15