[bfs序 线段树] HDU5957. Query on a graph

求出bfs序

那么一个点的儿子和孙子都是在bfs序上连续的一段

记一下这个区间，然后线段树维护一下

因为有环，那么肯定有多出来的一条边，给这条边的两个端点以及这两个端点连出去的点打个标记就好了

细节贼多

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <queue>

using namespace std;

typedef long long ll;

const int N=100010;

int n,t,q,cnt,m,G[N];
struct edge{
  int t,nx;
}E[N<<1];

inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &x){
  char c=nc(); x=0; int f=1;
  for(;c>'9'||c<'0';c=nc())if(c=='-')f=-f;for(;c>='0'&&c<='9';x=x*10+c-'0',c=nc()); x*=f;
}

inline void addedge(int x,int y){
  E[++cnt].t=y; E[cnt].nx=G[x]; G[x]=cnt;
  E[++cnt].t=x; E[cnt].nx=G[y]; G[y]=cnt;
}

int fa[N],vis[N],sl[N],sr[N],gsl[N],gsr[N],Q[N],pos[N],d1[N],d2[N];
int l,r;

inline void bfs(){
  vis[1]=1; Q[l=r=1]=1; pos[1]=1;
  while(l<=r){
    int x=Q[l++];
    for(int i=G[x];i;i=E[i].nx)
      if(!vis[E[i].t]){
    Q[++r]=E[i].t; pos[E[i].t]=r; fa[E[i].t]=x; vis[E[i].t]=1;
      }
  }
  for(int k=r;k;k--){
    int x=Q[k]; sl[x]=gsl[x]=1<<30; sr[x]=gsr[x]=0;
    for(int i=G[x];i;i=E[i].nx)
      if(fa[E[i].t]==x){
    sl[x]=min(sl[x],pos[E[i].t]); sr[x]=pos[E[i].t];
    gsl[x]=min(gsl[x],sl[E[i].t]); gsr[x]=max(sr[E[i].t],gsr[x]);
      }
  }
  for(int i=1;i<=n;i++) d1[i]=d2[i]=0;
  for(int k=2;k<=cnt;k++){
    int x=E[k].t,y=E[k^1].t;
    if(fa[x]==y || fa[y]==x) continue;
    d1[x]=y; d1[y]=x;
    for(int i=G[x];i;i=E[i].nx)
      if(E[i].t!=y) d2[E[i].t]=y;
    for(int i=G[y];i;i=E[i].nx)
      if(E[i].t!=x) d2[E[i].t]=x;
  }
}

void PutAns(ll x){
  if(x>=10) PutAns(x/10); putchar(x%10+'0');
}

ll tag[N<<2],tot[N<<2],size[N<<2];

void Build(int g,int l,int r){
  tag[g]=tot[g]=0; size[g]=r-l+1;
  if(l==r) return ;
  int mid=l+r>>1;
  Build(g<<1,l,mid); Build(g<<1|1,mid+1,r);
}


inline void add(int g,ll x){
  tag[g]+=x; tot[g]+=size[g]*x;
}

inline void Push(int g){
  if(!tag[g]) return ;
  add(g<<1,tag[g]); add(g<<1|1,tag[g]); tag[g]=0;
}

inline void Up(int g){
  tot[g]=tot[g<<1]+tot[g<<1|1];
}

void Modify(int g,int l,int r,int L,int R,ll x){
  if(l>r) return;
  if(l==L && r==R) return add(g,x);
  int mid=L+R>>1; Push(g);
  if(r<=mid) Modify(g<<1,l,r,L,mid,x);
  else if(l>mid) Modify(g<<1|1,l,r,mid+1,R,x);
  else Modify(g<<1,l,mid,L,mid,x),Modify(g<<1|1,mid+1,r,mid+1,R,x);
  Up(g);
}

ll Query(int g,int l,int r,int L,int R){
  if(l>r) return 0;
  if(l==L && r==R) return tot[g];
  int mid=L+R>>1; Push(g);
  if(r<=mid) return Query(g<<1,l,r,L,mid);
  else if(l>mid) return Query(g<<1|1,l,r,mid+1,R);
  else return Query(g<<1,l,mid,L,mid)+Query(g<<1|1,mid+1,r,mid+1,R);
}

int main(){
  read(t);
  while(t--){
    read(n); cnt=1;
    for(int i=1;i<=n;i++)
      G[i]=vis[i]=fa[i]=0;
    for(int i=1,x,y;i<=n;i++)
      read(x),read(y),addedge(x,y);
    bfs(); Build(1,1,n);
    read(m);
    for(int i=1;i<=m;i++){
      char opt; while((opt=nc())!='M' && opt!='Q');
      if(opt=='M'){
    int x,k,d;
    read(x); read(k); read(d);
    if(k==0) Modify(1,pos[x],pos[x],1,n,d);
    else if(k==1){
      Modify(1,pos[x],pos[x],1,n,d);
      Modify(1,sl[x],sr[x],1,n,d);
      if(fa[x]) Modify(1,pos[fa[x]],pos[fa[x]],1,n,d);
      if(d1[x]) Modify(1,pos[d1[x]],pos[d1[x]],1,n,d);
    }
    else{
      if(fa[x]){
        Modify(1,pos[fa[x]],pos[fa[x]],1,n,d);
        Modify(1,sl[fa[x]],sr[fa[x]],1,n,d);
        if(fa[fa[x]]) Modify(1,pos[fa[fa[x]]],pos[fa[fa[x]]],1,n,d);
      }
      else Modify(1,pos[x],pos[x],1,n,d);
      Modify(1,sl[x],sr[x],1,n,d);
      Modify(1,gsl[x],gsr[x],1,n,d);
      if(d1[x]){
        int y=d1[x];
        if(fa[y] && fa[x]!=fa[y] && fa[y]!=fa[fa[x]] && fa[fa[y]]!=fa[x]) Modify(1,pos[fa[y]],pos[fa[y]],1,n,d);
        if(fa[y]!=fa[x]) Modify(1,pos[y],pos[y],1,n,d);
        Modify(1,sl[y],sr[y],1,n,d);
      }
      if(d2[x] && fa[d2[x]]!=fa[x] && fa[fa[d2[x]]]!=x && fa[d2[x]]!=x) Modify(1,pos[d2[x]],pos[d2[x]],1,n,d);
    }
      }
      else{
    ll ans=0; int x,k;
    read(x); read(k);
    if(k==0) ans=Query(1,pos[x],pos[x],1,n);
    else if(k==1){
      ans=Query(1,pos[x],pos[x],1,n)+Query(1,sl[x],sr[x],1,n);
      if(fa[x]) ans+=Query(1,pos[fa[x]],pos[fa[x]],1,n);
      if(d1[x]) ans+=Query(1,pos[d1[x]],pos[d1[x]],1,n);
    }
    else{
      if(fa[x]){
        ans=Query(1,pos[fa[x]],pos[fa[x]],1,n)+Query(1,sl[fa[x]],sr[fa[x]],1,n);
        if(fa[fa[x]]) ans+=Query(1,pos[fa[fa[x]]],pos[fa[fa[x]]],1,n);
      }
      else ans=Query(1,pos[x],pos[x],1,n);
      ans+=Query(1,sl[x],sr[x],1,n);
      ans+=Query(1,gsl[x],gsr[x],1,n);
      if(d1[x]){
        int y=d1[x];
        if(fa[y] && fa[y]!=fa[x] && fa[y]!=fa[fa[x]] && fa[fa[y]]!=fa[x]) ans+=Query(1,pos[fa[y]],pos[fa[y]],1,n);
        if(fa[y]!=fa[x]) ans+=Query(1,pos[y],pos[y],1,n);
        ans+=Query(1,sl[y],sr[y],1,n);
      }
      if(d2[x] && fa[d2[x]]!=fa[x] && fa[fa[d2[x]]]!=x && fa[d2[x]]!=x) ans+=Query(1,pos[d2[x]],pos[d2[x]],1,n);
    }
    if(ans<0) putchar('-'),ans=-ans;
    PutAns(ans); putchar('\n');
      }
      //for(int i=1;i<=n;i++) printf("%lld ",Query(1,pos[i],pos[i],1,n)); putchar('\n');
    }
  }
  return 0;
}