1034 Head of a Gang

1034 Head of a Gang （30 point(s)）

题意：

​ 警察通过查看通话记录把一些人分成一个一个的帮派。

​ 判断是否是帮派的条件：

​ 1.ALLcost> K

​ 2.帮派成员大于2

​ 于是这道题可以用 有向图 来存储题目给的数据，然后DFS 遍历 的同时 求得连通分量（帮派个数),以及整个帮派的通话时间。

​ 由于题目给的每个顶点的信息是字符串。所以需要使用一个map<string,int> M;来映射name和G[ k ] 的k 的关系。

int dfs(int s){//dfs 遍历 head 和整个帮派的总通话时间
    if(vis[s]==true){return 0;}
    vis[s]=true;
    g_membercnt++;
    if(vex[s].cnt>g_Maxcost){
        g_Maxcost=vex[s].cnt;
        g_max_idx =s;
    }
    for(int i=0;i<peoplecnt;i++){
        g_allcost+=G[s][i];
        if(G[s][i]!=0&&vis[i]==false){
            dfs(i);
        }
    }
}

​ 而每个帮派的头头（head）的计算，则是在读入数据的时候就将每个顶点的通话时间都计算出来。DFS 的时候选出最大的那个就Ok了。

​ 最后输出的时候需要注意 是按照字典序输出所以sort 一下在输出。

AC代码

#include <vector>
#include <cstdio>
#include <map>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;
#define MAX 2001
int G[MAX][MAX]={0};
struct V{
    int cnt;
    string name;
};
V  vex[MAX];
vector<bool> vis;
map<string,int> M;//用map来映射 名字和 图中的序号的关系
map<string,int>::iterator g_it;

int peoplecnt =0;
int N,K;
int g_membercnt,g_Maxcost,g_max_idx,g_allcost;
//结果存储
struct res{
    const char* name;
    int member;
};
vector<res> head;
int dfs(int s);
bool cmp(const res&a,const res &b);
int main(int argc, char const *argv[])
{
    scanf("%d %d",&N,&K);
    for(int i=0;i<N;i++){
        string f,t;int cost;
        cin>>f>>t>>cost;
        g_it =M.find(f);
        if(g_it==M.end()){
            vex[peoplecnt].name=f;
            M[f] =peoplecnt;
            vis.push_back(false);
            peoplecnt++;
        }
        g_it =M.find(t);
        if(g_it==M.end()){
            vex[peoplecnt].name=t;
            vis.push_back(false);
            M[t] =peoplecnt;
            peoplecnt++;
        }
        G[M[f]][M[t]] =cost;
        vex[M[f]].cnt+=cost;
        vex[M[t]].cnt+=cost;
    }
    
    for(int i=0;i<peoplecnt;i++){
        if(vis[i]==false){
            g_membercnt =0;
            g_max_idx=i;
            g_Maxcost =0;
            g_allcost=0;
            dfs(i);
            if(g_membercnt>=3&&g_allcost>K){
                head.push_back(res{vex[g_max_idx].name.c_str(),g_membercnt});
            }
        }
    }
    sort(head.begin(),head.end(),cmp);
    printf("%d\n",head.size());
    for(int i=0;i<head.size();i++){
        printf("%s %d\n",head[i].name,head[i].member);
    }
    return 0;
}
int dfs(int s){//dfs 遍历 head 和整个帮派的总通话时间
    if(vis[s]==true){return 0;}
    vis[s]=true;
    g_membercnt++;
    if(vex[s].cnt>g_Maxcost){
        g_Maxcost=vex[s].cnt;
        g_max_idx =s;
    }
    for(int i=0;i<peoplecnt;i++){
        g_allcost+=G[s][i];
        if(G[s][i]!=0&&vis[i]==false){
            dfs(i);
        }
    }
}
bool cmp(const res&a,const res &b){
    return strcmp(a.name,b.name)<0?true:false;
}