第五天PAT-A1034 Head of a Gang并查集解法详细算法描述及代码解析

A1034

Description:

警方找犯罪团伙的头目的方式之一是检查人们的通话记录。如果AB之间有通话，则认为AB是相关的。关系的强度定义为两人通话的总时长。若一群人中存在两个人的的关系强度大于阈值K，则认为是一个团伙。对于每个团伙，总关系强度最大的人是头目。现在给出一组通话记录，找出这个团伙及其头目。

算法描述：

• 关于团伙人数，使用并查集，设置一个团伙人数记录数组(代码中为cnt)；
• 对于每条通话记录，为两者累加记录通话时间，为两者合并后得到的团伙增加总通话时长；
• 通话时间最长的人是团伙中的头目；
• 注意数组开的大小，一开始只在User数组开了2倍N，错了5组测试…心态直接崩了

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<string>
#include<algorithm>
#include<map>
#include<vector>
#include<queue>
#include<set>
using namespace std;
const int maxn = 2e3+5;
struct user{
   
    string name;
    int time;
    int id;
}User[maxn];
int n, k;   //通话个数即阈值
int mapin;  //姓名字符串 - ID映射下标
int fa[maxn];   //并查集father
map<string,int>m;   //姓名字符串 - ID映射
vector<int>cnt[maxn];   //记录每个团伙成员，存放下标对应ID的人所在团伙成员
set<string>ans;
int totaltime[maxn];    //团伙总时间

void init(){
       //初始化
    mapin = 0;  //姓名ID映射初始化
    m