1034. Head of a Gang (30)

http://www.patest.cn/contests/pat-a-practise/1034

好吧，这个题目还有好几个没过，要好好研究

主要用map，这是个神器啊！！！

貌似问题出现在回边访问，其实我没懂sad

#include <cstdio>
#include <map>
#include <vector>
#include <string>
#include <iostream>
//体中给了说最多1000条边，因此最多有2000个点
#define  MAX 2010
using namespace std;
struct edge{
int next;
int wei;
};
vector <edge> adj[MAX];
int weigh[MAX];
map <string,int> mp;
map <int,string> an;
map <string,int> gang;
bool vis[MAX];
void init(){
for(int i=0;i<MAX;i++){
adj[i].clear();
weigh[i]=0;
}
mp.clear();
an.clear();
}
void dfs(int x,int &ans,int &max,int &maxi){
vis[x]=true;
ans++;
if(weigh[x]>max){
maxi=x;
max=weigh[x];
}
for(int i=0;i<adj[x].size();i++){
int j=adj[x][i].next;
if(!vis[j]){
dfs(j,ans,max,maxi);
}
}
}
int main(){
freopen("in.txt","r",stdin);
int n,k;
while(scanf("%d %d",&n,&k)!=EOF){
init();
int cnt=0;
for(int i=0;i<n;i++){
int a,b;
string sa,sb;
int cost;
//scanf("%s%s%d",sa,sb,&cost);
cin>>sa>>sb>>cost;

map <string,int>::iterator it=mp.find(sa);
if(it==mp.end()){
a=cnt;
mp[sa]=cnt++;
}else{
a=it->second;
}
it=mp.find(sb);
if(it==mp.end()){
b=cnt;
mp[sb]=cnt++;
}else{
b=it->second;
}

for(it=mp.begin();it!=mp.end();it++){
an[it->second]=it->first;
}

weigh[a]+=cost;
weigh[b]+=cost;

edge tmp;
tmp.next=a;
tmp.wei=cost;
bool flag=false;
for(int i=0;i<adj[b].size();i++){
if(tmp.next==adj[b][i].next){
adj[b][i].wei+=tmp.wei;
flag=true;
break;
}
}
if(!flag){
adj[b].push_back(tmp);
}
flag=false;
tmp.next=b;
for(int i=0;i<adj[a].size();i++){
if(tmp.next==adj[a][i].next){
adj[a][i].wei+=tmp.wei;
flag=true;
break;
}
}
if(!flag){
adj[a].push_back(tmp);
}

}

//当每个点权都小于k，不是gang
int cnk=0;
for(int i=0;i<cnt;i++){
if(weigh[i]<=k){
cnk++;
}
}
if(cnk==cnt){
printf("0\n");
continue;
}

int res=0;
int ff[MAX];
for(int i=0;i<cnt;i++){
int ans=0;
int max=0;
int maxi=0;
if(!vis[i]){
dfs(i,ans,max,maxi);
if(ans>2){
ff[res++]=maxi;
for(int i=0;i<res;i++){
map <int,string>::iterator tt=an.find(ff[i]);
gang[tt->second]=ans;
}

}
}
}
printf("%d\n",res);
if(res!=0){
map <string,int>::iterator a=gang.begin();
for(;a!=gang.end();a++){
//printf("%s %d\n",a->first,a->second);
cout << a->first << ' '<< a->second <<endl;
}
}

}
return 0;
}