1034 Head of a Gang(7ms，并查集版，比图的版本快得多，更省内存

 

9ms，并查集版，比图的版本快得多，更更更省内存

题目

题意： 给出1000条以内的通话记录A B和权值w，和阈值k，如果一个团伙人数超过2人并且通话总权值超过k，令团伙里面的自身权值的最大值为头目，输出所有满足条件的团伙的头目，和他们团伙里面的人数

tip：并查集 + map映射 + sort

#include<iostream>
#include<vector>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
struct ss {
	int num;//真正的头头编号
	int count=1;//团伙人数
	int time=0;//每个人通话总时间
	int counttime=0;//总时间
};
map<string,int> nametonum;
map<int,string> numtoname;
map<int,struct ss> user;
int father[10003];
void init() {
	for(int i=0; i<10003; ++i)
		father[i]=i;
}
int find(int x) {
	return x==father[x]?father[x]:father[x]=find(father[x]);
}
void Uion(int a,int b) {
	int x=find(a);
	int y=find(b);
	if(x!=y)
		father[x]=y;
}
bool cmp(int a,int b) {
	return numtoname[user[a].num]<numtoname[user[b].num];
}
int main() {
	init();
	int n,m;
	cin>>n>>m;
	set<string> name;
	int c=0;
	for(int i=0; i<n; ++i) {
		string a,b;
		int time;
		cin>>a>>b>>time;
		if(name.find(a)==name.end()) {//登记a姓名
			name.insert(a);
			nametonum[a]=c;
			numtoname[c]=a;
			c++;
		}
		if(name.find(b)==name.end()) {//登记b姓名
			name.insert(b);
			nametonum[b]=c;
			numtoname[c]=b;
			c++;
		}
		user[nametonum[a]].time+=time;
		user[nametonum[b]].time+=time;
		Uion(nametonum[a],nametonum[b]);//建立联系
	}
	vector<int>master;//统计并查集头头 
	for(int i=0; i<n; ++i) {
		if(father[i]==i) {//找出并查集头头
			master.push_back(i);
			user[i].num=i;
		}
	}
	for(int i=0; i<n; ++i) {
		father[i]=find(i);
		//比较头头与其他成员通话时间时长，确定真正输出的头头
		user[father[i]].num=user[user[father[i]].num].time>user[i].time ?user[father[i]].num:i;
		if(father[i]!=i) {//算成员总数
			user[father[i]].count++;
		}
		user[father[i]].counttime+=user[i].time;//算群体内总通话时间
	}
	sort(master.begin(),master.end(),cmp);//对头头输出顺序排序
	vector<int>ans;
	for(int i=0; i<master.size(); ++i)
		if(user[master[i]].counttime>2*m&&user[master[i]].count>2)//gang成立条件
			ans.push_back(master[i]);
	cout<<ans.size()<<endl;//gang数量 
	for(int i=0; i<ans.size(); ++i)
		cout<<numtoname[user[ans[i]].num]<<" "<<user[ans[i]].count<<endl;
	return 0;
}