【NOI 2002】银河英雄传说（带权并查集）

最新推荐文章于 2019-11-22 19:41:10 发布

转载最新推荐文章于 2019-11-22 19:41:10 发布 · 129 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

原文链接：http://www.cnblogs.com/Patrickpwq/articles/9757874.html

本文介绍了一种使用带权并查集的数据结构解决特定问题的方法。通过维护节点到根节点的距离，可以高效地更新和查询飞船到队头的距离。在每次合并操作中，更新子树到新根节点的距离，并在查询时利用这一信息计算相对距离。

（小声bb 我觉得带权并查集比较玄妙）

传送门QAQ

很容易想到用并查集来维护

设size表示当前队列的数量（以i为队头的数量），to_root表示当前节点到祖宗的距离

则对于每一个飞船，它到队头的距离，就等于它到它祖先的距离加上它祖先到队头的距离，而它的祖先到队头的距离，也可以变成类似的。可以在getfather时顺便维护

哎。。还有多久才能停课啊。。真的不想上文化课了。。。。。

#include<bits/stdc++.h>
#define N 30005
using namespace std;
int father[N],to_root[N],size[N];
inline int getfather(int x)
{
    if(father[x]==x)    return x;
    int ff=father[x];
    father[x]=getfather(father[x]);
    to_root[x]+=to_root[ff];
    return father[x];
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL),cout.tie(NULL);
    for(int i=1;i<=30001;i++)    {father[i]=i;size[i]=1;}
    int T;
    cin>>T;
    while(T--)
    {
        char opt;
        int x,y,fx,fy;
        cin>>opt>>x>>y;
        fx=getfather(x),fy=getfather(y);
        if(opt=='M')
        {
            father[fx]=fy;
            to_root[fx]=size[fy];
            size[fy]+=size[fx];
            size[fx]=0;
        }
        if(opt=='C')
        {
            if(fx!=fy)    
            {
                cout<<"-1"<<'\n';
                continue;
            }
            cout<<abs(to_root[x]-to_root[y])-1<<'\n';
        }
    }
    return 0;
}