BZOJ3631&&洛谷P3258[JLOI2014]松鼠的新家

树上差分裸题

两个目标点之间的最短路径上的房间都要放一个糖，所以用树上差分，然后因为从一个房间出发是不需要糖果的，所以最后输出之前要减一再输出

看这里

代码

//By AcerMo
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int M=300500;
int n,a[M];
int dep[M],f[M][20],s[M];
int to[M*2],nxt[M*2],head[M],cnt;
inline int read()
{
	int x=0;char ch=getchar();
	while (ch>'9'||ch<'0') ch=getchar();
	while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
	return x; 
}
inline void write(int x)
{
	if (x<0) putchar('-'),x=-x;
	if (x>9) write(x/10);
	putchar(x%10+'0');
	return ;
}
inline void add(int x,int y)
{
	to[++cnt]=y;nxt[cnt]=head[x];head[x]=cnt;
	to[++cnt]=x;nxt[cnt]=head[y];head[y]=cnt;
	return ;
}
inline void dfs(int x,int fa)
{
	dep[x]=dep[fa]+1;f[x][0]=fa;
	for (int i=1;i<=18;i++)
	f[x][i]=f[f[x][i-1]][i-1];
	for (int i=head[x];i;i=nxt[i])
	if (to[i]!=fa) dfs(to[i],x);
	return ;
}
inline void lca(int x,int y)
{
	s[x]++;s[y]++;int l;
	if (dep[x]<dep[y]) swap(x,y);
	for (int i=18;i>=0;i--)
	if (dep[f[x][i]]>=dep[y])
	x=f[x][i];
	if (x==y) l=x;
	else 
	{
		for (int i=18;i>=0;i--)
		if (f[x][i]!=f[y][i])
		x=f[x][i],y=f[y][i];
		l=f[x][0];
	}
	s[l]--;s[f[l][0]]--;
	return ;
}
inline void getsum(int x)
{
	for (int i=head[x];i;i=nxt[i])
	if (to[i]!=f[x][0])
	getsum(to[i]),s[x]+=s[to[i]];	
	return ;
}
signed main()
{
	n=read();int x,y;
	for (int i=1;i<=n;i++) a[i]=read();
	for (int i=1;i<n;i++)
	x=read(),y=read(),add(x,y);
	dfs(1,0);
	for (int i=1;i<n;i++) lca(a[i],a[i+1]);
	getsum(1);s[a[1]]++;
	for (int i=1;i<=n;i++) s[a[i]]--;
	for (int i=1;i<=n;i++) write(s[i]),puts("");
	return 0;
}