hrbust 2371 GT’s Dream 【线段树】【并查集】

GT’s Dream

Time Limit: 1000 MS Memory Limit: 256000 K Total Submit: 83(31 users) Total Accepted: 29(21 users) Rating: Special Judge: No

Description

在现实中认了无数师傅却毫无长进的GT在梦中成为了某武侠世界的神。在这个世界中初始有n个人，他们各成一派。作为世界神GT总共会进行m次操作，每次操作有如下两种情况 1 x y 表示x所在的帮派吞并了y所在的帮派，若x与y本来就处于同一个帮派则该操作无效。 2 k 表示GT想要知道当前第k大的帮派有多少人，若当前帮派数量少于k个则输出-1。

Input

第一行输入一个T（T<=10）表示有T组测试数据。 每组数据的第一行包含两个数n,m（n<=100000,m<=300000）表示有n个人，m次操作 接下来m行包含每一次的操作情况。

Output

对于每一个2操作输出当前第k大的帮派有多少人。

Sample Input

1 5 4 1 2 4 1 3 4 2 1 2 2

Sample Output

3 1 并查集连接，线段树查询，这里只说一下线段树的的实现过程 线段树的叶子节点保存的是该节点编号的数值出现的次数，所以最开始只是1号节点为n，其他为0； 由于树上的节点只保存数字出现次数，所以在查询第K大值的时候先在右儿子节点查找，如果右儿子的数量大于k那么ans一定在右儿子里； 如果右儿子的值小于k，接着查询左儿子，因为右儿子里没有所以递归查询时，k更改为k-右儿子的值； 如果左儿子值也小于k，说明区间内没有第k大值直接return -1；

#include<bits/stdc++.h>
using namespace std;
#define lson rt << 1, l, mid
#define rson rt << 1 | 1, mid + 1, r
#define mms(x, y) memset(x, y, sizeof x)
#define MAX 100005
int n, m;
int pre[MAX], num[MAX], tree[MAX << 2];
int findx(int x)
{
    return x == pre[x] ? x : pre[x] = findx(pre[x]);
}
void PushUp(int rt)
{
    tree[rt] = tree[rt << 1] + tree[rt << 1 | 1];
}
void Update(int rt, int l, int r, int i, int v)
{
    if(l == r)
    {
        tree[rt] += v;
        return;
    }
    int mid = (l + r) >> 1;
    if(i <= mid)
        Update(lson, i, v);
    if(i > mid)
        Update(rson, i, v);
    PushUp(rt);
}
int Query(int rt, int l, int r, int i)
{
    if(l == r)
        return l;
    int mid = (l + r) >> 1;
    if(tree[rt << 1 | 1] - i >= 0)
        Query(rson, i);
    else if(tree[rt << 1] - i + tree[rt << 1 | 1] >= 0)
        Query(lson, i - tree[rt << 1 | 1]);
    else
        return -1;
}
void init()
{
    mms(num, 0);
    mms(tree, 0);
    scanf("%d%d", &n, &m);
    for(int i = 0; i <= n; i++)
    {
        pre[i] = i;
        num[i] = 1;
    }
    Update(1, 1, n, 1, n);
}
void join(int x, int y)
{
    int a = findx(x), b = findx(y);
    if(a != b)
    {
        pre[b] = a;
        Update(1, 1, n, num[a], -1);
        Update(1, 1, n, num[b], -1);
        num[a] += num[b];
        num[b] = 0;
        Update(1, 1, n, num[a], 1);
    }
}
int main()
{
    int N;
    scanf("%d", &N);
    while(N--)
    {
        init();
        while(m--)
        {
            int x, y, z;
            scanf("%d", &x);
            if(x == 1)
            {
                scanf("%d%d", &y, &z);
                join(y, z);
            }
            else
            {
                scanf("%d", &y);
                printf("%d\n", Query(1, 1, n, y));
            }
        }
    }
    return 0;
}