poj2895

时间:2021-05-02 12:25:07

题解:

splay,维护当前第k大

并查集维护当前集合

合并x,y时,del(num[x]),del(num[y]),insert(num[x]+num[y])

代码:

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=;
int pre[N],tot,xx,q,size[N],c[N][],fa[N],data[N],root,n,m,x,y,num[N];
void rot(int x)
{
int y=pre[x],k=(c[y][]==x);
size[y]=size[c[y][k]]+size[c[x][k]]+;
size[x]=size[c[x][!k]]+size[y]+;
c[y][!k]=c[x][k];
pre[c[y][!k]]=y;
pre[x]=pre[y];
if(pre[y])c[pre[y]][c[pre[y]][]==y]=x;
c[x][k]=y;pre[y]=x;
}
void splay(int x,int g)
{
for(int y=pre[x];y!=g;rot(x),y=pre[x])
if(pre[y]!=g)rot((x==c[y][])==(y==c[pre[y]][])?y:x);
if(g==)root=x;
}
void insert(int x)
{
int y=root;
while(c[y][x>data[y]]) y=c[y][x>data[y]];
data[++tot]=x;
c[tot][]=c[tot][]=;
pre[tot]=y;
if(y)c[y][x>data[y]]=tot;
splay(tot,);
}
void del(int x)
{
int y=root;
while(data[y]!=x) y=c[y][x>data[y]];
splay(y,);
y=c[root][];
bool b;
if(!y) b=,y=c[root][];else b=;
while(c[y][b]) y=c[y][b];
splay(y,root);
c[y][b]=c[root][b];pre[c[root][b]]=y;pre[y]=;root=y;
size[y]=size[c[y][!b]]+size[c[y][b]];
}
int find(int x)
{
if (x==fa[x])return x;
return fa[x]=find(fa[x]);
}
int findkth(int x)
{
int y=root;
while(x)
if(size[c[y][]]+<x)x-=size[c[y][]]+,y=c[y][];
else if(size[c[y][]]+==x) return data[y];
else y=c[y][];
return data[y];
}
int read()
{
char c;int x=;
for (;c<''||c>'';c=getchar());
for (;c>=''&&c<='';c=getchar())x=x*+c-;
return x;
}
void write(int x)
{
if (x>=)write(x/);
putchar(x%+);
}
int main()
{
n=read();m=read();
for (int i=;i<=n;i++)insert();
for (int i=;i<=n;i++)fa[i]=i,num[i]=;
while (m--)
{
xx=read();
if (xx==)
{
x=read();y=read();
int dx=find(x),dy=find(y);
if (dx!=dy)
{
fa[dx]=dy;
del(num[dy]);del(num[dx]);
num[dy]+=num[dx];
insert(num[dy]);
n--;
}
}
else
{
x=read();
write(findkth(n-x+)),puts("");
}
}
}

相关文章