码迷,mamicode.com
首页 > 其他好文 > 详细

并查集

时间:2016-09-06 15:13:52      阅读:176      评论:0      收藏:0      [点我收藏+]

标签:

刷题是遇到一题:http://www.nowcoder.com/questionTerminal/b1303e827e7f4df4a816598d008bbe72

 技术分享

这道题要用到并查集来解(Union-Find),这位网友总结的很详细:http://blog.csdn.net/dm_vincent/article/details/7655764

并查集主要包括两部分,find和union函数。其中find函数找根结点,union函数合并结点。根据初始化值不同,find和union函数的写法

也略有不同。

下面是某网友写的答案,略有瑕疵,我修改了下,加上点注释:

技术分享
int find(vector<int>& id, int i)
{
    //只有根结点或者未连接的点为负,其他点为其根结点的索引值
    //如果不是根结点,找到根结点
    while (id[i] >= 0)i = id[i];

    return i;
}

void unioned(vector<int>&id, int i, int j)
{

    //将根值更小的点设为根结点
    if (id[i] < id[j])
    {
        id[i] += id[j];
        id[j] = i;//直接保存根结点的索引
    }
    else
    {
        id[j] += id[i];
        id[i] = j;
    }
}
int main()
{
    string str;
    int count;//输入连通集个数
    while (cin >> str)
    {
        int len = str.length();
        //全部初始化为-1.
        vector<int> id(len, -1);
        cin >> count;
        for (int i = 0; i < count; i++)
        {
            int p, q;
            cin >> p >> q;
            int pRoot = find(id, p);
            int qRoot = find(id, q);
            if (pRoot != qRoot)
                unioned(id, pRoot, qRoot);//注意:是根结点相连接
        }
        vector<vector<int>> mat(len);
        for (int i = 0; i < id.size(); i++)
        {
            if (id[i] == -1)continue;

            int root = find(id, i);
            mat[root].push_back(i);//保存各个连通集的位置索引(按根结点)
        }
        //
        for (int i = 0; i < mat.size(); i++)
        {
            vector<char> chs;//保存连通集上的字符
            //如果不是根结点,则直接跳过
            for (int j = 0; j < mat[i].size(); j++)
            {
                chs.push_back(str[mat[i][j]]);
            }
            sort(chs.begin(), chs.end());//将同个连通集上的字符排序
            //写回字符串中
            for (int j = 0; j < mat[i].size(); j++)
                str[mat[i][j]] = chs[j];
        }
        cout << str;
    }
}
View Code

 

并查集

标签:

原文地址:http://www.cnblogs.com/573177885qq/p/5845576.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!