该题依旧是最小生成树问题,只是有一点小小的变化。。。。。做完这题,猛然发现为什么prim算法优先于Kursal了。。。。。
#include<iostream> #include<string.h> #include<algorithm> #include<limits.h> #define N 105 #define FOR(i,s,t) for(int i=(s);i<=t;++i) using namespace std; int n,m; int father[N]; struct edge{int x,y,len; int flag; }map[5005]; typedef struct edge Node; int cmp(const void *a,const void *b) { return ((Node*)a)->len>((Node*)b)->len?1:-1; } void make_set() { FOR(i,1,n) father[i]=i; } int find_set(int a) { while(a!=father[a]) a=father[a]; return a; } void Kursal() { int sum=0,ans=0; FOR(i,1,m) { int x=find_set(map[i].x); int y=find_set(map[i].y); if(x!=y) { father[y]=x; if(!map[i].flag) sum+=map[i].len; ans++; } if(ans==n-1) break; } cout<<sum<<endl; } int main() { while(cin>>n&&n) { m=n*(n-1)/2; make_set(); FOR(i,1,m) {cin>>map[i].x>>map[i].y>>map[i].len>>map[i].flag; if(map[i].flag) map[i].len=0; } qsort(map+1,m,sizeof(Node),cmp); Kursal(); } return 0; }