PAT甲级1034 Head of a Gang【bfs】

题目：https://pintia.cn/problem-sets/994805342720868352/problems/994805456881434624

题意：

给定n条记录（注意不是n个人的记录），两个人之间的关系的权值为这两个人之间所有电话记录的时间之和。

一个连通块的权值为所有关系权值之和。

如果一个连通块节点数大于2，且权值大于给定的k，称这是一个gang，拥有关系权值和最多的人是gang的头。

要求输出gang的数量，每个gang的头，每个gang的人数。按照gang的头的字典序排序。

思路：

bfs求连通块。有几个注意点：

1、给定的是n条记录，$n<=1000$， 这说明人数应该是2000以内而不是1000以内。否则会有段错误

2、每一条记录都要考虑，bfs时不能仅判断这个节点是否被访问，还应该设置边的vis数组。

#include<cstdio>
#include<cstdlib>
#include<map>
#include<set>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath> 
#include<stack>
#include<queue>

#define inf 0x7fffffff
using namespace std;
typedef long long LL;
typedef pair<string, string> pr;

int n, k;
map<string, int>mp;
map<int, string>revmp;
const int maxn = 2005;
int g[maxn][maxn]; 
int tot = 0;

struct node{
    int head;
    int mxweight = -1;
    int num;
}gang[maxn];
int ans = 0;
bool vis[maxn];
bool evis[maxn][maxn];

void bfs()
{
    for(int i = 0; i < tot; i++){
        if(!vis[i]){
            int weight = 0;
            queue<int>que;
            que.push(i);
            vis[i] = true;
            //printf("
%d ", i);
            
            while(!que.empty()){
                int now = que.front();que.pop();
                gang[ans].num++;
                int res = 0;
                for(int j = 0; j < tot; j++){
                    res += g[now][j];
                    if(!evis[now][j] && g[now][j]){
                        weight += g[now][j];
                        evis[now][j] = evis[j][now] = true;
                    }
                    if(!vis[j] && g[now][j]){
                        que.push(j);
                        vis[j] = true;
                    }
                }
                if(res > gang[ans].mxweight){
                    gang[ans].mxweight = res;
                    gang[ans].head = now;
                }
            }
            if(weight > k && gang[ans].num > 2){
                ans++;
            }
            else{
                gang[ans].num = 0;
                gang[ans].mxweight = -1;
                gang[ans].head = 0;
            }
        }
    }
}


bool cmp(node a, node b)
{
    return revmp[a.head] < revmp[b.head];
}

int main()
{
    scanf("%d%d", &n, &k);
    for(int i = 0; i < n; i++){
        string a, b;
        int val;
        cin>>a>>b>>val;
        
        if(mp.find(a) == mp.end()){
            revmp[tot] = a;
            mp[a] = tot++;
        }
        if(mp.find(b) == mp.end()){
            revmp[tot] = b;
            mp[b] = tot++;
        }
        g[mp[a]][mp[b]] += val;
        g[mp[b]][mp[a]] += val;
    }
    //cout<<tot<<endl;
//    for(int i = 0; i < tot; i++){
//        for(int j = 0; j < tot; j++){
//            printf("%d ", g[i][j]);
//        }
//        printf("
");
//    }
    bfs();
    printf("%d
", ans);
    sort(gang, gang + ans, cmp);
    for(int i = 0; i < ans; i++){
        cout<<revmp[gang[i].head]<<" "<<gang[i].num<<endl;
    }
    return 0;
}