bzoj1103 [POI2007]大都市meg

傳送門

本來是爲了刷dfs序的題目的，但是用了樹鏈剖分做。。。
題目輸入解釋有點問題，但是應該能理解

CODE：

#include<cstdio>
const int N=250005;
const int INF=1e9;
struct edge
{
    int nxt,to;
}a[N<<2];
int t[N<<2];
int head[N],size[N],f[N],son[N],pos[N],deep[N],top[N];
int n,m,x,y,num,tot;
inline int max(const int &a,const int &b){return a>b?a:b;}
inline int min(const int &a,const int &b){return a<b?a:b;}
inline void swap(int &a,int &b){a^=b,b^=a,a^=b;}
inline void add(int x,int y)
{
    a[++num].nxt=head[x],a[num].to=y,head[x]=num;
    a[++num].nxt=head[y],a[num].to=x,head[y]=num;
}
void dfs(int now,int fa,int depth)
{
    deep[now]=depth;
    f[now]=fa;
    size[now]=1;
    int tmp=-INF;
    for(int i=head[now];i;i=a[i].nxt)
      if(a[i].to!=fa)
      {
        dfs(a[i].to,now,depth+1);
        size[now]+=size[a[i].to];
        if(size[a[i].to]>tmp) tmp=size[a[i].to],son[now]=a[i].to;
      }
}
void dfs2(int now,int high)
{
    top[now]=high;
    pos[now]=++tot;
    if(son[now]) dfs2(son[now],high);
    for(int i=head[now];i;i=a[i].nxt)
      if(a[i].to!=f[now]&&a[i].to!=son[now]) dfs2(a[i].to,a[i].to);
}
inline void update(int now)
{
    t[now]=t[now<<1]+t[now<<1|1];
}
void build(int l,int r,int now)
{
    if(l==r)
    {
        if(l!=1) t[now]=1;
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,now<<1);
    build(mid+1,r,now<<1|1);
    update(now);
}
void Add(int p,int l,int r,int now)
{
    if(l==r)
    {
        t[now]=0;
        return;
    }
    int mid=(l+r)>>1;
    if(p<=mid) Add(p,l,mid,now<<1);
    else Add(p,mid+1,r,now<<1|1);
    update(now);
}
int ask(int L,int R,int l,int r,int now)
{
    if(L<=l&&r<=R) return t[now];
    int mid=(l+r)>>1,ans=0;
    if(L<=mid) ans=ask(L,R,l,mid,now<<1);
    if(R>mid) ans+=ask(L,R,mid+1,r,now<<1|1);
    return ans;
}
int askpath(int x,int y)
{
    int L,R;
    if(top[x]==top[y])
    {
        L=min(pos[x],pos[y]);
        R=max(pos[x],pos[y]);
        return ask(L,R,1,n,1);
    }
    if(deep[top[x]]<deep[top[y]]) swap(x,y);
    L=min(pos[x],pos[top[x]]);
    R=max(pos[x],pos[top[x]]);
    int ans=ask(L,R,1,n,1);
    return ans+askpath(y,f[top[x]]);
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
      scanf("%d%d",&x,&y),add(x,y);
    dfs(1,0,1);
    dfs2(1,1);
    build(1,n,1);
    scanf("%d",&m);
    m+=n-1;
    while(m--)
    {
        char c=getchar();
        while(c!='W'&&c!='A') c=getchar();
        if(c=='W') scanf("%d",&x),printf("%d\n",askpath(1,x));
        else scanf("%d%d",&x,&y),f[y]==x?Add(pos[y],1,n,1):Add(pos[x],1,n,1);
    }
    return 0;
}

反思：以後一定要注意樹鏈剖分在修改的時候一定是修改的pos[x]而不是x！