pat 甲级 Head of a Gang

这道题思路，首先将所有的联系看作无向图。答案就是要找出符合条件(点的数量大于2，且总的时间大于阈值k）的连通图。

#include<iostream>
#include<algorithm>
#include<string>
#include<vector>
#include<unordered_map>

using namespace std;

int n,k;
unordered_map<string,vector<pair<string,int>> > g; //first存放的是名字，second存放的是有关联的名字以及对话时间。
unordered_map<string,int> total;//first存放名字，second则是这个人的总的时间
unordered_map<string,bool> st;//first存放名字，second看是否访问过，类似于st[]数组，只是下标是字符串，故用哈希代替
int dfs(string ver,vector<string>& nodes){//深搜，nodes存放着同一个连通图的节点，即同一个gang
    st[ver]=true;//已经访问过
    nodes.push_back(ver);/同一个gang
    int sum=0;
    for(auto item:g[ver]){
        sum+=item.second;
        string cur=item.first;//相关联的对象
        if(!st[cur]) sum+=dfs(cur,nodes);//未被访问，递归深搜
    }
    return sum;
}
int main()
{
    cin>>n>>k;
    while(n--){
        string a,b;
        int t;
        cin>>a>>b>>t;
        g[a].push_back({b,t});//无向图是有两条边的有向图，故分两次存储。
        g[b].push_back({a,t});
        total[a]+=t;
        total[b]+=t;
    }
    vector<pair<string,int>> res;//存放答案，
    for(auto item:total){
        string node=item.first;
        vector<string> nodes;
        int sum=dfs(node,nodes)/2;//因为是无向图，一条边计算了两次，所以需要/2
        if(nodes.size()>2&&sum>k){//符合题目条件
            string boss=node;
            for(auto cur:nodes){//寻找head
                if(total[cur]>total[boss])
                    boss=cur;
            }
            res.push_back({boss,nodes.size()});
        }
    }
    sort(res.begin(),res.end());
    cout<<res.size()<<endl;
    for(auto temp:res){
        cout<<temp.first<<" "<<temp.second<<endl;
    }
    return 0;
}