HDOJ--1879--继续畅通工程

继续畅通工程

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 17919 Accepted Submission(s): 7726



Problem Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。



Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。

当N为0时输入结束。


Output
每个测试用例的输出占一行，输出全省畅通需要的最低成本。


Sample Input

3
1 2 1 0
1 3 2 0
2 3 4 0
3
1 2 1 0
1 3 2 0
2 3 4 1
3
1 2 1 0
1 3 2 1
2 3 4 1
0

Sample Output

3
1
0

思路：这道题也是用最小生成树来计算的，嗯，我用的是prime算法，一会儿再用kruskal算法再做一遍。
ac代码：

#include<stdio.h>
#include<string.h>
#define INF 0x3f3f3f3f
int n,vis[101],map[101][101],dis[101];
int sum;
void prime(){
	int i;
	sum=0;
	memset(vis,0,sizeof(vis));
	for(i=1;i<=n;i++)
		dis[i]=map[1][i];
	vis[1]=1;
	for(i=1;i<n;i++){
		int temp=INF,k,j;
		for(j=1;j<=n;j++)
			if(vis[j]==0&&dis[j]<temp)
				temp=dis[k=j];
		sum+=temp;
		vis[k]=1;
		for(j=1;j<=n;j++)
			if(vis[j]==0&&dis[j]>map[k][j]){
				dis[j]=map[k][j];
			}
						
	}
}
int main(){
	while(scanf("%d",&n)!=EOF&&n!=0){
		memset(map,INF,sizeof(map));
		for(int i=1;i<=n*(n-1)/2;i++){
			int a,b,cost,flag;
			scanf("%d%d%d%d",&a,&b,&cost,&flag);
			if(flag==1)
				map[a][b]=map[b][a]=0;
			else
				map[a][b]=map[b][a]=cost;
		}
		prime();
		printf("%d\n",sum);
	}
	return 0;
}

接下来就是用kruskal算法做出来的。（人不能懒啊，我就是因为觉得稍微改一下就好了 把hdoj1863给稍微改了下结果一直wa，还是自己再敲一遍比较有用啊）
ac代码：

#include<stdio.h>
#include<algorithm>
using namespace std;
int n,m,per[101*101];
struct node{
	int b,e,cost,flag;
}edge[101*101];
bool cmp(node c,node d){
	if(c.flag==d.flag)
		return c.cost<d.cost;
	else
		return c.flag>d.flag;
}
void init(){
	for(int i=1;i<=m;i++)
		per[i]=i;
}
int find(int x){
	int r=x;
	while(r!=per[r]){
		r=per[r];
	}
	int i=x,j;
	while(i!=r){
		j=per[i];
		per[i]=r;
		i=j;
	}
	return r;
}
int join(int a,int b){
	int fa=find(a);
	int fb=find(b);
	if(fa!=fb){
		per[fa]=fb;
		return 1;
	}
	else
		return 0;
}
int main(){
	while(scanf("%d",&n)!=EOF&&n!=0){
		m=n*(n-1)/2;
		init();
		int sum=0,i;
		for(i=1;i<=m;i++)
			scanf("%d%d%d%d",&edge[i].b,&edge[i].e,&edge[i].cost,&edge[i].flag);
		sort(edge+1,edge+m+1,cmp);
		for(i=1;i<=m;i++){
			if(join(edge[i].b,edge[i].e)){
				if(edge[i].flag==1)
					sum+=0;
				else
					sum+=edge[i].cost;
			}
		}
		printf("%d\n",sum);
	}
	return 0;
}