[POI2007]MEG-Megalopolis 树状数组 + dfs序前缀和好题

最新推荐文章于 2025-08-05 19:33:31 发布

EM-LGH

最新推荐文章于 2025-08-05 19:33:31 发布

阅读量197

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构

本文链接：https://blog.youkuaiyun.com/liyong1009s/article/details/81987023

本文介绍了一种使用区间树进行二分图匹配的方法，通过构建特定的图结构和区间树，实现了对节点间匹配状态的有效管理和快速查询。文章详细解释了区间树的构建、更新和查询操作，并通过一个具体的算法实现展示了如何在二分图中进行匹配过程的高效处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=250000+6;
int C[N<<1],head[N<<1],to[N<<1],nex[N<<1],idx[N],cnt,idx2[N];
int nums=0;
void add_edge(int u,int v){
    nex[++cnt]=head[u],head[u]=cnt,to[cnt]=v;
}
int lowbit(int t){
    return t&(-t);
}
void update(int t,int delta){
    while(t<(N<<1))C[t]+=delta,t+=lowbit(t);
}
int query(int t){
    int tmp=0;
    while(t>0)tmp+=C[t],t-=lowbit(t);
    return tmp;
}
void dfs(int u,int fa)
{
    ++nums;
    if(u!=1)update(nums,1);
    idx[u]=nums;
    for(int v=head[u];v;v=nex[v])
        if(to[v]!=fa)dfs(to[v],u);
    ++nums;
    if(u!=1)update(nums,-1);
    idx2[u]=nums;
}
int main()
{
   // freopen("in.txt","r",stdin);
    int n,m;
    scanf("%d",&n);
    for(int i=1;i<n;++i){
        int a,b;
        scanf("%d%d",&a,&b);
        add_edge(a,b);
        add_edge(b,a);
    }
    dfs(1,-1);
    scanf("%d",&m);
    for(int i=1;i<=m+n-1;++i)
    {
        char s[4];
        scanf("%s",s);
        if(s[0]=='W')
        {
            int a;
            scanf("%d",&a);
            printf("%d\n",query(idx[a]));
        }
        else if(s[0]=='A')
        {
            int a,b;
            scanf("%d%d",&a,&b);
            int cur=max(idx[a],idx[b]);
            int cur2=min(idx2[a],idx2[b]);
            update(cur,-1);
            update(cur2+1,1);
        }
    }
    return 0;
}