POJ 1797 条件最大生成树

这题做的囧死，看了样例很兴奋的写了个dijkstra交上去WA，重新看了下题目，发现是每条路都有一个最大承载量，好吧，我又想错了，发现不是网络流，因为样例用最大流说不通，再仔细看了一遍题目，发现是叫你找一条路使得单条路的流量最大，果断写了个最大生成树，觉得最大生成树的最小边就是解，叫上去，再次WA。

 

想了一会儿才发现不对，如果有一条路流量很大，但是这条路建出来的时候生成树没有建好呢？后来在Kruskal函数里加了一句很重要的话，果断AC。。。。。

 

本题的思路就是求最大生成树，当1和n在一个集合里的时候就返回最小边权，这个边权就是解。

 

我的代码：

#include <cstdio> #include <cstdlib> #include <cstring> #include <algorithm> using namespace std; const int MAX=1100; const int oo=0x3f3f3f3f; struct Edge{ int u,v,len; bool operator<(const Edge& e) const{ return len>e.len; } void read(){ scanf("%d%d%d",&u,&v,&len); } }e[MAX*MAX]; int p[MAX],rank[MAX],n,m; void init(){ for(int i=1;i<=n;i++){ p[i]=i; rank[i]=0; } } int find(int x){ return p[x]=x==p[x]?p[x]:find(p[x]); } void Link(int x,int y){ if(rank[x]>rank[y]){ p[y]=x; }else{ p[x]=y; if(rank[x]==rank[y]){ rank[y]++; } } } bool merge(int x,int y){ int px=find(x); int py=find(y); if(px==py){ return false; } Link(px,py); return true; } int Kruskal(){ int ret=0; int cnt=1; init(); sort(e,e+m); for(int i=0;i<m&&cnt<n;i++){ if(merge(e[i].u,e[i].v)){ cnt++; ret=e[i].len; } if(find(1)==find(n)){ break; } } return ret; } int main(){ int t,cnt=1; bool done=false; scanf("%d",&t); while(t--){ if(done){ puts(""); }else{ done=true; } scanf("%d%d",&n,&m); for(int i=0;i<m;i++){ e[i].read(); } printf("Scenario #%d:/n",cnt++); printf("%d/n",Kruskal()); } return 0; }