PAT 甲级 1034 PAT Ranking 个人错误总结

有点难，编码复杂。
dfs需要找到，每个连通图数值总数，最大值结点，结点个数，还要用hash来修正字母，最后还要sort。
很容易写错，一开始我想的是用并查集，特别麻烦，因为没法保证根节点就是最大点，也没法记录总数，只能在最后用map遍历。

#include<bits/stdc++.h>
using namespace std;
vector<int> llist[30000];
int v[30000];
struct node{
	string a;
	int n;
};
int stn(string a){
	int s=0;
	for(int i=0;i<3;i++){
		s=a[i]-'A'+s*26;
	}
	return s;
}
string nts(int a){
	string str;
	str+=(char)(a/(26*26)+'A');
	str+=(char)(a%(26*26)/26+'A');
	str+=(char)(a%26+'A');
	return str;
}
bool flag[30000];
int vmaxn=0;
int tag;
int sum=0;
int cnt=0;
void dfs(int root){
	if(v[root]>vmaxn){
		vmaxn=v[root];
		tag=root;
	}
	cnt++;
	sum+=v[root];
	flag[root]=true;
	for(int i=0;i<llist[root].size();i++){
		if(flag[llist[root][i]]==false)dfs(llist[root][i]);
	}
}
bool cmp(node a,node b){
	return a.a<b.a;
}
int main(){
	int n,k,temp;
	cin>>n>>k;
	string s1,s2;
	int n1,n2;
	for(int i=0;i<n;i++){
		cin>>s1>>s2>>temp;
		n1=stn(s1);
		n2=stn(s2);
		v[n1]+=temp;
		v[n2]+=temp;
		llist[n1].push_back(n2);
		llist[n2].push_back(n1);
	}
	vector<node> ans;
	for(int i=0;i<26*26*26;i++){
		if(llist[i].size()&&flag[i]==false){
			vmaxn=0;
			sum=0;
			cnt=0;
			dfs(i);
			if(sum>k*2&&cnt>2){
				node tt;
				tt.a=nts(tag);
				tt.n=cnt;
				ans.push_back(tt);
			}
		}
	}
	cout<<ans.size()<<"\n";
	sort(ans.begin(),ans.end(),cmp);
	for(int i=0;i<ans.size();i++){
		cout<<ans[i].a<<" "<<ans[i].n<<"\n";
	}
    return 0;
}