题意:
有一群猫,现在给它们编号,然后组队,0 1 2 代表 1 号和 2 号的队伍合并。 1 4 询问第 4 大的队伍 Size 是多大。
解:(贴的,他写的好。表示学习)
线段树。建立一个线段树。每段记录group的大小在l与r之前的数目。初始时大小为1的组有N个,大小不为1的组均为0个。所以对于每段,若左边界为1,则其数据域为N,否则为0。以后不断合并group的过程中,设两个group的带大小分别为k1,k2。则线段树中包含k1或k2的结点数据域均减1,而包含k1+k2的结点数据域加1。因为合并后,原来大小为k1,k2的组已经不存在,取而代之的是新的大小为k1+k2的组。以后查询第k大的组则比较简单:在线段树上若未到达叶子结点,k若小于或等于右子树的数据域,则从右子树继续查询第k大的组;否则从左子树查询第k-s大的组,其中s为右子树的数据域,因为在右子树中已经有s个组比要查询的组大了,在左子树查询第k-s大的组即为所有组中第k大的组。
做题过程:
并查集那里和初始化那里出了一点问题。 并查集的话需要判断一下他们是不是在同一个集合,如果是在同一个集合,就不用合并了。
/*
Pro: 0
Sol:
date:
*/
#include<cstdio>
#include<cstring>
const int maxn = 201001;
#define lson l, m , rt << 1
#define rson m + 1, r, rt << 1 | 1
#define havem int m = (l + r) >> 1
using namespace std;
int sum[maxn << 2],p[maxn],num[maxn],n,Q;
void build(int l, int r, int rt){
if(l == 1) sum[rt] = n;
else sum[rt] = 0; //这里错了一次,初始化问题
if(l == r) return ;
havem;
build(lson); build(rson);
}
void update(int pos,int cc, int l, int r, int rt){//单点更新,凡是包含pos的节点都要改变
sum[rt] += cc;
if(l == r) return ; havem;
if(pos <= m) update(pos,cc,lson);
else update(pos,cc,rson);
}
int query(int k, int l, int r, int rt){//单点询问
if(l == r) return l; havem;
if(sum[rt << 1 | 1] >= k) return query(k,rson);
else return query(k - sum[rt << 1 | 1],lson);
}
int find(int x){
return x==p[x] ? x : p[x]=find(p[x]);
}
int main(){
scanf("%d%d",&n,&Q);
for(int i = 1; i <= n; i ++){
num[i] = 1; p[i] = i;
}
int op,a,b;
build(1,n,1);
for(int i = 1;i <= Q; i ++){
scanf("%d",&op);
if(op == 0){
scanf("%d%d",&a,&b);
int x = find(a);
int y = find(b);
if(x == y) continue ;//这里需要判断一下
update(num[x],-1,1,n,1);
update(num[y],-1,1,n,1);
update(num[x] + num[y],1,1,n,1);
num[x] += num[y];
p[y] = x;
}else{
scanf("%d",&a);
printf("%d\n",query(a,1,n,1));
}
}
return 0;
}