第六章图论3 AcWing 1518. 团伙头目-CSDN博客

该博客介绍了如何使用图论中的BFS或DFS以及并查集解决团伙头目问题。在无向图中，通过计算每个连通块的总权值，判断是否超过指定阈值k，并找出帮派头领。代码实现中，首先读取图的数据结构，然后遍历每个节点，利用dfs进行深度优先搜索，找到连通块并计算其大小和总权值，筛选出符合条件的团伙并记录其头目。最后，按团伙规模排序并输出结果。

第六章图论3 AcWing 1518. 团伙头目

原题链接

AcWing 1518. 团伙头目

算法标签

图论图的遍历 BFS DFS 并查集

思路

在这里插入图片描述
由于图是无向图，存储过程中存储为两条有向边。因此实际总权和=计算总权和 / 2（相当于每条边被重复计算两次，正向计算一次，反向计算一次）
再判断是否总权和>k且点数>2 若是，表明该连通块为一个帮派，遍历该连通块中每个点，找出权值最大值（即为帮派头领），最后排序输出

代码

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define int long long
#define x first
#define y second
#define ump unordered_map
#define pq priority_queue
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>=b;--i)
using namespace std;
typedef pair<int, int> PII;
const int N = 10005;
//int t, n, m, cnt, ans; 
int n, k;
ump<string, vector<pair<string, int>>> g;
ump<string, int> to;
ump<string, bool> st;
inline int rd(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}
int dfs(string v, vector<string>& nodes){
    st[v]=true;
    nodes.push_back(v);
    int sum=0;
    for(auto ed:g[v]){
        sum+=ed.y;
        string cu=ed.x;
        if(!st[cu]){
            sum+=dfs(cu, nodes);
        }
    }
    return sum;
}

signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>k;
	rep(i, 0, n){
	    string n1, n2;
	    int t;
	    cin>>n1>>n2>>t;
	    g[n1].push_back({n2, t});
	    g[n2].push_back({n1, t});
	    to[n1]+=t;
	    to[n2]+=t;
	}
	vector<pair<string, int>> res;
    for(auto i:to){
        string v=i.x;
        vector<string> nodes;
        int sum=dfs(v, nodes)/2;
        if(nodes.size()>2&&sum>k){
            string bo=nodes[0];
            for(string node:nodes){
                if(to[bo]<to[node]){
                   bo=node;
                }
            }
            res.push_back({bo, nodes.size()});
        }
    }	
    sort(res.begin(), res.end());
    cout<<res.size()<<"\n";
    for(auto i:res){
        cout<<i.x<<" "<<i.y<<"\n";
    }
	return 0;
}