pat-a1034. Head of a Gang (30)

一开始的方向都错了。。这个题对边进行深搜，开始一直对点进行深搜的。在边搜索的同时，把边的权重放进两个节点的权重数组里面。同时用深搜确定有多少组织。每个组织有哪些成员存起来。权重和大于阈值的输出头领以及人数就行了。

刚开始第3测试用例一直段错误。。查了半天才反应过来1000条通话记录最多能有2000人。。数组开小了。。

#include<iostream>
#include<string>
#include<map>
#include<vector>
#include<algorithm>
#include<functional>
using namespace std;
struct gang{
string name;
int n;
gang(string a,int b):name(a),n(b){
}
};
vector<gang> vg;
map<string,int>code;
map<int,string>encode;
map<int,vector<int> >tree;
int visit[2010];
int w[2010][2010];
int v[2010];
int k=0,ans=0,p=1;
bool cmd(const gang& a,const gang& b){
return a.name<b.name;
}
void dfs(int t){
if(visit[t]==0) tree[ans].push_back(t);
int va=0;
visit[t]=1;
for(int i=1;i<p;++i){
if(w[t][i]){
v[t]+=w[t][i];
v[i]+=w[t][i];
w[t][i]=w[i][t]=0;
if(visit[i]==0){
tree[ans].push_back(i);
visit[i]=1;
}
dfs(i);
}
}
}
int main(){
int n;
cin>>n>>k;
string n1,n2;
int t;
while(n--){
cin>>n1>>n2>>t;
if(code[n1]==0){
code[n1]=p;
encode[p]=n1;
p++;
}
if(code[n2]==0){
code[n2]=p;
encode[p]=n2;
p++;
}
int f=code[n1];
int to=code[n2];
w[f][to]+=t;
w[to][f]+=t;
}
for(int i=1;i<p;++i){
if(visit[i]==0){
dfs(i);
ans++;
}
}
int ans1=0;
for(int i=0;i<ans;++i){
int ma=0;
int pos=0,temp=0;
int len=tree[i].size();
for(int j=0;j<len;++j){
temp+=v[tree[i][j]];
if(ma<v[tree[i][j]]){
ma=v[tree[i][j]];
pos=tree[i][j];
}
}
temp/=2;
if(temp>k&&len>2){
ans1++;
vg.push_back(gang(encode[pos],len));
}
}
cout<<ans1<<endl;
if(ans1){
sort(vg.begin(),vg.end(),cmd);
for(int i=0;i<ans1;++i) cout<<vg[i].name<<' '<<vg[i].n<<endl;
}
return 0;
}