题意:
有一棵树,n个节点,从a1出发->a2->a3->…->an,(a为一个排列),每经过一个节点时,该节点加一,问最终每个节点的值
思路:
类似差分,lca(u,v)+1,u-1,v-1,从上往下统计答案就可以了u+1,v+1,lca(u,v)-1,fa[lca[u,v]]-1,从下往上扫描
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
const int N=3e5+100;
const int DEG=20;
struct Edge{
int to,next;
}e[N*2];
int a[N],ans[N],tot,head[N];
void init(){
tot=0;
memset(head,-1,sizeof(head));
}
void addedge(int from,int to){
e[tot]=(Edge){to,head[from]};
head[from]=tot++;
}
int fa[N][DEG];
int deg[N],degree[N];
void bfs(int root){
queue<int>Q;
deg[root]=0;
fa[root][0]=0;
Q.push(root);
while(!Q.empty()){
int tmp=Q.front();
Q.pop();
for(int i=1;i<DEG;i++)
fa[tmp][i]=fa[fa[tmp][i-1]][i-1];
for(int i=head[tmp];i!=-1;i=e[i].next){
int v=e[i].to;
if(v==fa[tmp][0]) continue;
degree[tmp]++;
deg[v]=deg[tmp]+1;
fa[v][0]=tmp;
Q.push(v);
}
}
}
int LCA(int u,int v){
if(deg[u]>deg[v]) swap(u,v);
int tu=u,tv=v;
for(int det=deg[v]-deg[u],i=0;det;det>>=1,i++)
if(det&1)
tv=fa[tv][i];
if(tu==tv) return tu;
for(int i=DEG-1;i>=0;i--){
if(fa[tu][i]==fa[tv][i])
continue;
tu=fa[tu][i],tv=fa[tv][i];
}
return fa[tu][0];
}
void get_ans(int n){
queue<int>Q;
for(int i=1;i<=n;i++)
if(degree[i]==0)
Q.push(i);
while(!Q.empty()){
int u=Q.front();
Q.pop();
for(int i=head[u];i!=-1;i=e[i].next){
int v=e[i].to;
if(v!=fa[u][0]) continue;
degree[v]--,ans[v]+=ans[u];
if(degree[v]==0)
Q.push(v);
}
}
}
int main(){
int n,u,v;
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
init();
for(int i=1;i<n;i++){
scanf("%d%d",&u,&v);
addedge(u,v),addedge(v,u);
}
bfs(1);
for(int i=2;i<=n;i++){
int lca=LCA(a[i],a[i-1]);
ans[ a[i-1] ]++,ans[ fa[a[i]][0] ]++;
ans[ fa[lca][0] ]--,ans[ lca ]--;
}
get_ans(n);
for(int i=1;i<=n;i++)
printf("%d\n",ans[i]);
return 0;
}