bzoj 3631 松鼠的新家

来源：互联网发布：淘宝网牛仔裤女款编辑：程序博客网时间：2024/04/27 10:55

挺难得有没有。。

~~查了查题解~~

没有看太明白区间++–，遂放弃

决定使用树剖+线段树+lazy标记。

嗯所以代码量有点可观。。。。。

#include <cstdio>  #include <cstring>  #include <iostream>  #include <algorithm>  using namespace std;  #define lson pos<<1  #define rson pos<<1|1  #define N 300005  int n,a[N],ans[N];  int cnt,head[N],next[2*N],to[2*N];  int now,size[N],deep[N],fa[N],son[N],p_id[N],id_p[N],top[N];  int f[4*N];  void add(int x,int y)  {      to[++cnt]=y;      next[cnt]=head[x];      head[x]=cnt;  }  void dfs(int x,int d)  {      size[x]=1;deep[x]=d;      for(int y,i=head[x];i;i=next[i])          if((y=to[i])!=fa[x])          {              fa[y]=x;              dfs(y,d+1);              if(size[y]>size[son[x]])                  son[x]=y;              size[x]+=size[y];          }  }  void create(int x,int d)  {      p_id[x]=++now;id_p[now]=x;      id_p[now]=x;      top[x]=d;      if(son[x])          create(son[x],d);      for(int y,i=head[x];i;i=next[i])          if((y=to[i])!=fa[x]&&y!=son[x])              create(y,y);  }  void lazy(int pos)  {      f[lson]+=f[pos];      f[rson]+=f[pos];      f[pos]=0;  }  void fix(int pos,int l,int r,int x,int y)  {      if(x<=l&&r<=y){ f[pos]++; return; }      if(f[pos]) lazy(pos);      int mid=(l+r)>>1;      if(y<=mid)          fix(lson,l,mid,x,y);      else if(x>mid)          fix(rson,mid+1,r,x,y);      else         fix(lson,l,mid,x,y),          fix(rson,mid+1,r,x,y);  }  void getans(int pos,int l,int r)  {      if(l==r){ ans[id_p[l]]=f[pos]; return; }      if(f[pos]) lazy(pos);      int mid=(l+r)>>1;      getans(lson,l,mid);      getans(rson,mid+1,r);  }  void work(int x,int y)  {      int f1=top[x],f2=top[y];      while(f1!=f2)      {          if(deep[f1]<deep[f2])              swap(x,y),swap(f1,f2);          fix(1,1,n,p_id[f1],p_id[x]);          x=fa[f1];          f1=top[x];      }      if(deep[x]>deep[y])          swap(x,y);      fix(1,1,n,p_id[x],p_id[y]);  }  int main()  {      cin>>n;      for(int i=1;i<=n;i++)          scanf("%d",&a[i]);      for(int x,y,i=1;i<n;i++)          scanf("%d%d",&x,&y),          add(x,y),add(y,x);      dfs(1,1);      create(1,1);      for(int i=2;i<=n;i++)          work(a[i],a[i-1]);      getans(1,1,n);      for(int i=1;i<=n;i++)          printf("%d\n",i==a[1]?ans[i]:ans[i]-1);      return 0;  }

0 0