LA4730 Kingdom

1.题目描述：点击打开链接

2.解题思路：本题利用BIT+并查集解决。根据题意可以发现，x值是没有意义的，因此本题实际上是一维的一个操作题。由于line操作需要输出州的个数和城市的个数，且是统计一个点处的结果，自然可以想到用离线标记法来解决，又因为查询操作也比较多，因此可以用BIT来标记和查询。那么如何来更新一个州的城市数目呢？可以用并查集解决，但是有一个问题：两个连通分量合并成一个连通分量时候，怎么处理先前标记过的那些区间？根据题意可知，一个州至少有2个城市，而且一个州可以用[l,r]来准确描述，因此可以事先删除连通分量1的区间和连通分量2的区间，合并后，添加上更大的连通分量区间即可。这样，本题得以完美解决。

注意：因为本题maxn比较大，因此这里的BIT的起点在maxn，终点在1位置。那么相应的修改操作变成x-=lowbit(x),x+=lowbit(x)。

3.代码：

#include<iostream>
#include<algorithm>
#include<cassert>
#include<string>
#include<sstream>
#include<set>
#include<bitset>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<cctype>
#include<complex>
#include<functional>
#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

#define rep(i,n) for(int i=0;i<(n);i++)
#define me(s) memset(s,0,sizeof(s))
#define pb push_back
#define lid (id<<1)
#define rid (id<<1|1)

typedef long long ll;
typedef pair<int,int> P;

const int maxn=1000000+10;
const int maxm=100000+10;

struct Node
{
    int l,r;
    int pa;
    int city;
}tr[maxm]; //每个结点就是一个州

int a[maxn],b[maxn]; //a数组标记州的个数，b数组标记城市个数

int lowbit(int x){return x&-x;}
int findset(int x){return tr[x].pa!=x?x:tr[x].pa=findset(tr[x].pa);}

void add(int l,int r,int state,int city) //给区间(l,r]都添加上state，city
{
    if(l>=r)return;
    while(r>0)
    {
        a[r]+=state;
        b[r]+=city;
        r-=lowbit(r);
    }
    while(l>0)
    {
        a[l]-=state;
        b[l]-=city;
        l-=lowbit(l);
    }
}

void query(int x,int&ans1,int&ans2)//查询点x处的state个数，city个数
{
    ans1=ans2=0;
    while(x<maxn)
    {
        ans1+=a[x];
        ans2+=b[x];
        x+=lowbit(x);
    }
}

void Union(int u,int v) //并查集合并操作
{
    int x=findset(u),y=findset(v);
    if(x!=y)
    {
        if(tr[x].city!=1)  //首先删除x,y所属的并查集所在的区间
            add(tr[x].l,tr[x].r,-1,-tr[x].city);
        if(tr[y].city!=1)
            add(tr[y].l,tr[y].r,-1,-tr[y].city);
        tr[y].pa=x;
        tr[x].city+=tr[y].city;
        tr[x].l=min(tr[x].l,tr[y].l);
        tr[x].r=max(tr[x].r,tr[y].r);
        add(tr[x].l,tr[y].r,1,tr[x].city); //合并后，添加上新的连通分量的区间
    }
}

int main()
{
    int T;
    int n,m;
    char str[20];
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=0;i<n;i++)
        {
            scanf("%d%d",&m,&tr[i].r);
            tr[i].pa=i,tr[i].city=1,tr[i].l=tr[i].r;
        }
        scanf("%d",&m);
        int x,y,ans1,ans2;
        for(int i=0;i<maxn;i++)
            a[i]=b[i]=0;
        while(m--)
        {
            scanf("s%d%d",str,&x,&y);
            if(str[0]=='r')Union(x,y);
            else
            {
                query(x+1,ans1,ans2); //由于是左开右闭区间，因此查询点x+1处的结果
                printf("%d %d\n",ans1,ans2);
            }
        }
    }
}