GT’s Dream | ||||||
| ||||||
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;
}