題意：

給出n次通話記錄，當通話的人數超過2人并且通話總時長超過k時，這些人就是犯罪團伙，其中通話時間最大的人是頭目，要求按字母序輸出頭目和團伙人數

思路：

就是并查集，將兩兩通話的人關聯在一起，因為通話的人的姓名是按三位字母給出的，所以要離散化，并且對應要用一些STL保存映射。我自己寫的時候有兩個點沒過，后來改過了，錯誤原因寫在下面注釋里了。網上的代碼也有用DFS找連通分量做的。

#include<iostream> #include<string> #include<vector> #include<map> #include<set> #include<algorithm> using namespace std; const int INF = 0xfffff; const int maxn = 2010; //這里開1005會出現段錯誤,因為是n個通話可能有2n個人map<string, int> gang; map<int, string> gangName; int gangTime[maxn]; int p[maxn]; int n, k,pos=0; vector<int> head[maxn]; set<string> res;void init() {for (int i = 0; i < maxn; i++) {p[i] = i;head[i].push_back(i);} }int find(int x) {if (x == p[x]) return x;return p[x] = find(p[x]); }void merge(int x, int y) {x = find(x);y = find(y);if (x != y) {p[x] = y;for (auto i = head[x].begin(); i != head[x].end(); i++) {head[y].push_back(*i);}head[x].clear(); //這里必須清零} }int main() {scanf("%d%d", &n, &k);init();for (int i = 0; i < n; i++) {string name1, name2;int time;cin >> name1 >> name2 >> time;if (gang.find(name1) == gang.end()) {gang[name1] = pos;gangName[pos++] = name1;}if (gang.find(name2) == gang.end()) {gang[name2] = pos;gangName[pos++] = name2;}gangTime[gang[name1]] += time;gangTime[gang[name2]] += time;merge(gang[name1], gang[name2]);}for (int i = 0; i < pos; i++) {if (head[i].size() > 2) {int maxTime = -1, index;int totalTime = 0;for (auto x : head[i]) {if (maxTime < gangTime[x]) {maxTime = gangTime[x];index = x;}totalTime += gangTime[x];}if (totalTime > 2*k) {res.insert(gangName[index]);}}}if (res.empty())cout << 0 << endl;else {cout << res.size() << endl;for (auto x : res) {int h = find(gang[x]);cout << x << " " << head[h].size() << endl;}}return 0; }

?

轉載于:https://www.cnblogs.com/seasonal/p/10343631.html

總結

以上是生活随笔為你收集整理的PAT1034 Head of a Gang (30)(并查集)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。