PAT(Advanced) 1034 Head of a Gang C++实现 并查集 路径压缩

PAT(Advanced) 1034 Head of a Gang 并查集 路径压缩

题目链接

1034 Head of a Gang

题目大意

根据通话记录来找到帮派的头目，一个帮派由三个及以上的人组成且其成员之间通话时长大于阈值K

算法思路

根据题意，一个帮派的人在同一个连通分量中，通过并查集，累加连通分量的关系权值和连通分量中的顶点数，找到符合条件的输出头目和人数，并查集合并时将关系权重较大的顶点作为根结点，需要注意的是找到连通分量所属的根结点每次都需要调用find()，否则某些情况会出错

根据题意，输出的名字需要进行递增排序，根据std::map的定义，std::map低层由红黑树实现，迭代器遍历得到的结果即为递增有序的结果。同时，若不要求有序，使用std::unordered_map效率相对较高，根据其定义，std::unordered_map低层由哈希表实现，时间复杂度O(1)，远低于std::map的O(n)

实现代码

/*
author : eclipse
email  : eclipsecs@qq.com
time   : Mon Oct 12 21:03:16 2020
*/
#include <bits/stdc++.h>
using namespace std;

unordered_map<string, string> root;
string head;

string find(string name) {
	// 并查集带路径压缩
    return root[name] == name ? name : root[name] = find(root[name]);
}

void unionRoot(string name1, string name2, unordered_map<string, int>& talkingTime) {
	// 合并
    string root1 = find(name1);
    string root2 = find(name2);
    // 关系权值较大者作为帮派头目
    if (talkingTime[root1] > talkingTime[root2]) {
        root[root2] = root[root1];
    } else {
        root[root1] = root[root2];
    }
}

int main(int argc, char const *argv[]) {
    int N, K;
    while (~scanf("%d%d", &N, &K)) {
        string name1, name2;
        int time;
        unordered_map<string, int> talkingTime;
        unordered_map<string, vector<string> > calls;
        root.clear();
        for (int i = 0; i < N; i++) {
            cin >> name1 >> name2 >> time;

			// 记录每个人的通话时间
            if (talkingTime.count(name1) == 0) {
                talkingTime[name1] = 0;
            }
            if (talkingTime.count(name2) == 0) {
                talkingTime[name2] = 0;
            }

			// 并查集， 初始化为自身
            if (root.count(name1) == 0) {
                root[name1] = name1;
            }
            if (root.count(name2) == 0) {
                root[name2] = name2;
            }

			// 累计每个人的通话时间
            talkingTime[name1] += time;
            talkingTime[name2] += time;
			
			// 彼此加入图中
            calls[name1].push_back(name2);
            calls[name2].push_back(name1);
        }
		
		// 每条通话记录的两人，合并
        for (unordered_map<string, vector<string> >::iterator it = calls.begin(); it != calls.end(); it++) {
            for (vector<string>::iterator name = it->second.begin(); name != it->second.end(); name++) {
                unionRoot(it->first, *name, talkingTime);
            }
        }

        map<string, pair<int, int> > gangs;

		// 找到所有连通分量并累计关系权值和人数
        for (unordered_map<string, string>::iterator it = root.begin(); it != root.end(); it++) {
            string gangHead = find(it->second);
            if (gangs.count(gangHead) == 0) {
                gangs[gangHead] = make_pair(0, 0);
            }
            gangs[gangHead].first++;
            gangs[gangHead].second += talkingTime[it->first];
        }

        vector<pair<string, int> > answer;
        // 找到满足要求的帮派
        for (map<string, pair<int, int> >::iterator it = gangs.begin(); it != gangs.end(); it++) {
            if (it->second.first > 2 && it->second.second / 2 > K) {
                answer.push_back(make_pair(it->first, it->second.first));
            }
        }
		
		// 输出答案
        printf("%d\n", answer.size());
        for (vector<pair<string, int> >::iterator it = answer.begin(); it != answer.end(); it++) {
            cout << it->first << " " << it->second << endl;
        }
    }
    return 0;
}

样例输入1

8 59
AAA BBB 10
BBB AAA 20
AAA CCC 40
DDD EEE 5
EEE DDD 70
FFF GGG 30
GGG HHH 20
HHH FFF 10

样例输出1

2
AAA 3
GGG 3

样例输入2

8 70
AAA BBB 10
BBB AAA 20
AAA CCC 40
DDD EEE 5
EEE DDD 70
FFF GGG 30
GGG HHH 20
HHH FFF 10

样例输出2

0

鸣谢

PAT

最后

• 由于博主水平有限，不免有疏漏之处，欢迎读者随时批评指正，以免造成不必要的误解！