题178.pat甲级练习-1034 Head of a Gang (30 分)

---

题178.pat甲级练习-1034 Head of a Gang (30 分)

---

一、题目

二、题解

本题要你找出总通话时间大于K，并且圈子人数大于2人的圈子，输出每个符合条件的圈子中通话时间最长的人以及圈子的人数，并注意输出顺序按名字字典序。本质就是个并查集问题，难度不大，就是想的时候感觉要照顾的信息有点多，又要并查得到圈子人数以及圈子的总通话时间又要知道每个圈子里的人这样才方便找圈子中最长通话时长的人。为此，我直接无脑将set放在了每个人的信息里，用于存储那个人身为根时，那个圈子里所有的人，这样可以通过set.size()得到圈子人数并且遍历set就可以找到那个圈子最长通话时长的人。

#include <bits/stdc++.h>

using namespace std;

const int maxsize=2001;

int N,K;

struct Ifo
{
    int weight;//表示那个人总的通话时长
    int weightsum;//身为根时表示那个圈子总的通话时长
    string parent;
    set<string> s;//身为根时用于存储那个圈子里所有的人
};

map<string,Ifo> m;

string findRoot(string name)//找根
{
    if(m[name].parent=="")
    {
        return name;
    }
    else
    {
        return m[name].parent=findRoot(m[name].parent);//路径压缩
    }
}

void unionSet(string name1,string name2,int weight)//并集
{
    string root1=findRoot(name1);
    if(root1==name1)//如果当前根为自己的话需要先将身为根时对应的set--那个圈子中加入自己，不然那个set可还是空的呢
    {
        m[root1].s.insert(name1);
    }
    string root2=findRoot(name2);
    if(root2==name2)
    {
        m[root2].s.insert(name2);
    }
    if(root1!=root2)
    {
        if(m[root1].s.size()>m[root2].s.size())//按秩归并
        {
            m[root2].parent=root1;
            m[root1].weightsum+=m[root2].weightsum;//将小圈的总的通话时长加到大圈里
            m[root1].weightsum+=weight;//将本次输入的通话时长加到大圈
            for(auto it=m[root2].s.begin(); it!=m[root2].s.end(); it++)//将小圈的人加到大圈里
            {
                m[root1].s.insert(*it);
            }
        }
        else
        {
            m[root1].parent=root2;
            m[root2].weightsum+=m[root1].weightsum;
            m[root2].weightsum+=weight;
            for(auto it=m[root1].s.begin(); it!=m[root1].s.end(); it++)
            {
                m[root2].s.insert(*it);
            }
        }
    }
    else//如果已经是同一个圈了那么只用将本次输入的通话时长加进去就好
    {
        m[root1].weightsum+=weight;
    }
    return;
}

int main()
{
    cin>>N>>K;
    for(int i=0; i<N; i++)
    {
        string name1,name2;
        int weight;
        cin>>name1>>name2>>weight;
        unionSet(name1,name2,weight);
        m[name1].weight+=weight;//更新那个人的通话时长
        m[name2].weight+=weight;
    }
    int count0=0;
    vector<string> res;
    for(auto it=m.begin(); it!=m.end(); it++)
    {
        if(it->second.parent=="")
        {
            if(it->second.s.size()>2&&it->second.weightsum>K)//满足圈子人数大于2并且总的通话时长大于阈值K的才算上
            {
                count0++;
                res.push_back(it->first);
            }
        }
    }
    if(count0==0)
    {
        cout<<0<<endl;
    }
    else
    {
        cout<<count0<<endl;
        map<string,int> num;//装要输出的东西
        for(int i=0; i<res.size(); i++)
        {
            string ans;
            int maxweight=0;
            for(auto it=m[res[i]].s.begin(); it!=m[res[i]].s.end(); it++)
            {
                if(m[*it].weight>maxweight)
                {
                    maxweight=m[*it].weight;
                    ans=*it;
                }
            }
            num[ans]=m[res[i]].s.size();
        }
        for(auto it=num.begin(); it!=num.end(); it++)//没办法，由于要名字字典序输出所以得把它们集中再排序输出（不然测试点2过不去），所以干脆放map里了
        {
            cout<<it->first<<" "<<it->second<<endl;
        }
    }
    return 0;
}

---