Description
Input
Output
Sample Input
1 4 5 3 2
1 2
2 4
2 3
4 5
Sample Output
2
1
2
1
HINT
2<= n <=300000
话说前几天搞地理历史会考没发过题解……现在想起考试都是痛
这题……我也不知道该说什么
原来我是当树链剖分的模板打的
结果样例wa了……调半天调不出来
然后对着标程调出来了……结果数据一测T了
仔细一看n<=30w……TMD原来树链剖分是骗分
正解是这样的
我们注意到在树上的修改之间不存在在线的询问,就是说,可以用离线的做法搞
考虑对于树上一条链x->y修改,可以直接把x->root的+1、把fa[y]->root的-1
其实就是差分啦
然后修改变O(1)的了……最后从叶节点到根dp一下就完了
注意修改x->lca和y->lca的时候lca被找了两次,要去掉
……因为dfs1的时候以1而不是a[1]为根还wa了一次
#include<cstdio> #include<iostream> #include<cstring> #include<cstdlib> #include<algorithm> #include<cmath> #include<queue> #include<deque> #include<set> #include<map> #include<ctime> #define LL long long #define inf 0x7fffffff #define N 300010 using namespace std; inline LL read() { LL x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } struct edge{int to,next;}e[2*N]; int n,cnt; int a[N],f[N],bin[20]; int head[N],dep[N],fa[N][20]; inline void ins(int u,int v) { e[++cnt].to=v; e[cnt].next=head[u]; head[u]=cnt; } inline void insert(int u,int v) { ins(u,v); ins(v,u); } inline void dfs1(int x,int d) { dep[x]=d; for (int i=1;i<20;i++) if (d>=bin[i]) { fa[x][i]=fa[fa[x][i-1]][i-1]; }else break; for (int i=head[x];i;i=e[i].next) if (fa[x][0]!=e[i].to) { fa[e[i].to][0]=x; dfs1(e[i].to,d+1); } } inline int LCA(int a,int b) { if (dep[a]<dep[b])swap(a,b); int res=dep[a]-dep[b]; for (int i=0;i<20;i++) if (res & bin[i])a=fa[a][i]; for (int i=19;i>=0;i--) if (fa[a][i]!=fa[b][i]) { a=fa[a][i]; b=fa[b][i]; } if (a==b)return a; return fa[a][0]; } inline void dfs2(int x) { for (int i=head[x];i;i=e[i].next) if (e[i].to!=fa[x][0]) { dfs2(e[i].to); f[x]+=f[e[i].to]; } } int main() { n=read(); bin[0]=1;for (int i=1;i<20;i++)bin[i]=bin[i-1]*2; for (int i=1;i<=n;i++)a[i]=read(); for (int i=1;i<n;i++) { int x=read(),y=read(); insert(x,y); } dfs1(a[1],0); for (int i=1;i<n;i++) { int lca=LCA(a[i],a[i+1]); f[a[i]]++;f[a[i+1]]++; f[lca]--; f[fa[lca][0]]--; } dfs2(a[1]); for(int i=2;i<=n;i++)f[a[i]]--; for (int i=1;i<=n;i++)printf("%d\n",f[i]); }