题解 P2713 【罗马游戏】

· · 题解

加了修饰的左偏树模板题

模板题链接,双倍经验

简简单单把左偏树模板打上就行了,注意数据范围(n<=1e6,m<=1e5)。

以上为扯淡内容,正文开始。

题意(翻译成术语):

对n个小根堆做m个操作:

M (x,y)合并x所在堆和y所在堆(如果x或y被删除则忽略次操作);

K(x)输出x所在堆的堆顶元素并将其删除,如果x已经被删除这输出0 ;

百度百科:左偏树。

外节点:没有左子树或没有右子树的节点;

点的距离:点到它后代中离它最近的外节点经过的边数;

合并方法(小根堆):

设要将堆顶元素为 Y 的堆并入堆顶元素为 X 的堆(谁并入谁无所谓,这里是为了便于解释),则将X与Y比较,如果X>Y就交换X与Y,将Y继续与X的右子节点比较(递推)。

同时左偏树要保证往左偏,即保证左子节点的距离要比右子节点的距离要大。

删除最小的点(堆顶元素)的方法:

标记堆顶元素已删除,合并其左子树和右子树;

话不多说,上代码:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,a[1000010],d[1000010],ls[1000010],rs[1000010],f[1000010],x,y;//a是每个节点的值,d是节点的距离,ls左子节点,rs右子节点,f父节点(并查集)
bool b[1000010];//是否已经删除
char z;
int find(int x)
{
    if(f[x]==x)
    return x;
    return f[x]=find(f[x]);
}//并查集
int merge(int x,int y)//合并(返回当前节点的值)
{
    if(!x||!y)//有一棵树已经空了
    return x+y;//返回
    if(a[x]>a[y]||(a[x]==a[y]&&x>y))//交换条件
    swap(x,y);
    rs[x]=merge(rs[x],y);//右子节点
    if(d[ls[x]]<d[rs[x]])//如果左子节点距离小于右子节点
    swap(ls[x],rs[x]);//交换左右子树
    d[x]=d[rs[x]]+1;//算出当前点的距离
    return x;
}
int main()
{
    cin>>n;
    d[0]=-1;
    for(int i=1;i<=n;i++)
    {
        f[i]=i;
        cin>>a[i];
    }
    cin>>m; 
    while(m--)
    {
        cin>>z;
        if(z=='M')
        {
            cin>>x>>y;
            if(b[x]||b[y])
            continue;//有一个点已经被删除了
            x=find(x);
            y=find(y);
            if(x==y)//已经在同一个堆中了
            continue;
            f[x]=f[y]=merge(x,y);
        }
        else if(z=='K')
        {
            cin>>x;
            if(b[x])//已经删除
            {
                cout<<0<<endl;
                continue;
            }
            x=find(x);
            cout<<a[x]<<endl;
            b[x]=1;//标记删除
            f[x]=f[ls[x]]=f[rs[x]]=merge(ls[x],rs[x]);//合并左右子树
            ls[x]=rs[x]=d[x]=0;
        }
    }
    return 0;
}