[BZOJ1103][POI2007]大都市meg（dfs序+bit||链剖）

题目描述

传送门

题解

链剖裸题对不对吖~
然而ATP大爷觉得写模板题太low了，于是教我写bit。
用dfs序将一个点拆成两个点，在in点加正权，out点加负权，这样查询一个点到根路径上的权就是差这个点in的前缀和。辣样的话用一个bit维护就可以了。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 250005

int n,m,a,b,dfs_clock,ans;
int tot,point[N],nxt[N*2],v[N*2];
int C[N*2],in[N*2],out[N*2];

void addedge(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void add(int loc,int val)
{
    for (int i=loc;i<=2*n;i+=i&(-i))
        C[i]+=val;
}
int query(int loc)
{
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i))
        ans+=C[i];
    return ans;
}
void dfs(int x,int fa)
{
    in[x]=++dfs_clock;
    if (x!=1) add(dfs_clock,1);
    for (int i=point[x];i;i=nxt[i])
        if (v[i]!=fa)
            dfs(v[i],x);
    out[x]=++dfs_clock;
    if (x!=1) add(dfs_clock,-1);
}
int main()
{
    scanf("%d",&n);
    for (int i=2;i<=n;++i)
    {
        scanf("%d%d",&a,&b);
        addedge(a,b);
    }
    dfs(1,0);
    scanf("%d",&m);
    for (int i=1;i<=n+m-1;++i)
    {
        char opt=getchar();
        while (opt!='A'&&opt!='W') opt=getchar();
        if (opt=='A')
        {
            scanf("%d%d",&a,&b);
            if (in[a]<in[b]) swap(a,b);
            add(in[a],-1); add(out[a],1);
        }
        else
        {
            scanf("%d",&a);
            ans=query(in[a]);
            printf("%d\n",ans);
        }
    }
}

总结

①这个dfs序+bit的思路非常有用！感谢ATP神犇！