UVALive 4730 -树状数组+带权并查集

题意：有T组测试数据，每组数据的N表示有N个城市，接下来的N行里每行给出每个城市的坐标(0<=x,y<=1000000)，然后有M(1<M<200000)个操作，操作有两类，

（1）"road A B"，表示将城市A和城市B通过一条道路连接，如果A和B原来属于不同的城市群，经过这个操作，A和B就在一个城市群里了，保证每条道路不会和其他道路相交（除了端点A和B）。

（2）"line C"，表示查询当穿过y=C的直线，有多少个城市群、这几个城市群一共有多少个城市。

以C建立一颗线段树，

 每次 road A,B 操作，会新生成一个洲，那么对于这个洲，我们用并查集维护其城市数量num,和其最下、最上延伸范围sonl,sonr，也就是说这个新的洲会覆盖 【sonl,sonr】这个范围，那么我们在一颗树状数组里 这个范围的 洲数量 add 1， 另一个树状数组里（管辖城市数量的），add num

ps:当前生成之前的，A,B所在的洲要先抹去掉。

对于查询line C，直接分别在两棵树分别查询两个信息即可

范围有点大，离散化一下即可。

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <queue>
#include <map>
#include <set>
#include <vector>
using namespace std;
#define  inf 0x7fffffff
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
int n,m;
class Tree
{
    int tree[100131+50];
public:
    void init()
    {
        memset(tree,0,sizeof tree);
    }
    int lowbit(int x)
    {
        return x&-x;
    }
    void add(int x,int value)
    {
        for (int i=x; i<=100000; i=i+lowbit(i))
        {
            tree[i]+=value;
        }
    }
    int get(int x)
    {
        int sum=0;
        for (int i=x; i; i-=lowbit(i))
        {
            sum+=tree[i];
        }
        return sum;
    }

};
Tree tp1,tp2;
int fa[100132];
int sonl[100132];
int sonr[100131];
int num[100132];
int find(int x)
{
    if (x==fa[x]) return x;
    else  return fa[x]=find(fa[x]);
}
int city[100132][2];
void Union(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if (fx==fy) return;
    tp1.add(sonl[fx],-1);
    tp1.add(sonr[fx]+1-1,1);
    tp1.add(sonl[fy],-1);
    tp1.add(sonr[fy]+1-1,1);
    tp2.add(sonl[fx],-num[fx]);
    tp2.add(sonr[fx]+1-1,num[fx]);
    tp2.add(sonl[fy],-num[fy]);
    tp2.add(sonr[fy]+1-1,num[fy]);

    sonl[fy]=min(sonl[fy],sonl[fx]);
    sonr[fy]=max(sonr[fy],sonr[fx]);
    num[fy]+=num[fx];
    fa[fx]=fy;
    tp1.add(sonl[fy],1);
    tp1.add(sonr[fy]+1-1,-1);
    tp2.add(sonl[fy],num[fy]);
    tp2.add(sonr[fy]+1-1,-num[fy]);
    return  ;
}
int bb[100132];
int idx[100132];
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        tp1.init();
        tp2.init();
        cin>>n;
        int x,y;
        char tmp[15];

        for (int i=1; i<=n; i++)
            scanf("%d%d",&city[i][0],&city[i][1]);
        for (int i=1; i<=n; i++ ) bb[i]=city[i][1];
        sort(bb+1,bb+1+n);
        int cun=unique(bb+1,bb+1+n)-bb-1;
        for (int i=1;i<=n;i++)
        {
            int id=lower_bound(bb+1,bb+1+cun,city[i][1])-bb;
            idx[i]=id;
        }
        for (int i=1; i<=n; i++) fa[i]=i,sonl[i]=sonr[i]=idx[i],num[i]=1;
        int m;
        cin>>m;
        for (int i=1; i<=m; i++)
        {
            scanf("%s",tmp);
            if (tmp[0]=='r')
            {
                scanf("%d%d",&x,&y);
                x++,y++;
                Union(x,y);
            }
            else
            {
                scanf("%d.5",&x);
                x=upper_bound(bb+1,bb+1+cun,x)-bb;
                x--;
                int ans1=tp1.get(x);
                int ans2=tp2.get(x);
                printf("%d %d\n",ans1,ans2);
            }
        }
    }


    return 0;
}