HDOJ 1879 继续畅通工程 最小生成树 kruskal && prim

继续畅通工程

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

Total Submission(s): 17846    Accepted Submission(s): 7683

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
  
  

注-此题为： HDOJ 1879 继续畅通工程

思路：    将已连通的看做一个整体，将没连通的存入结构体，根据 克鲁斯卡尔  思想，求最低成本

例如：

第二组测试数据数据：

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

        2 和 3  有路，看做整体，将（1     2     1     0）（1     3     2     0）存入结构体，按成本从小到大排序

由于 2  和  3  已有路，不用再修  ，看做一个整体，还有一个  1  ，   1  到这个整体有两种  1 和 2 ，1 和 3；

1 和 2  成本为 1  ；1 和 3  ， 成本为  2；  所以要修 1 和 2  ，即 总成本 为 1  。

已AC代码：(kruscal)

#include<cstdio>
#define MAX 10010
#include<algorithm>
using namespace std;

int n,m;
int per[125];   // 并查集 

struct node{
	int u,v,w;   //w为成本 
}s[MAX];

bool cmp(node a,node b)
{
	return a.w<b.w;
}

void into()   //初始化 
{
	for(int i=0;i<=n;++i)
		per[i]=i;
}

int find(int x)     // 查找根节点 
{
	if(x==per[x])
		return x;
	return find(per[x]);
}

bool join(int a,int b)    //合并根节点 并判断是否成环 
{
	int fa=find(a);
	int fb=find(b);
	if(fa!=fb)
	{
		per[fa]=fb;
		return true;
	}
	return false;
}

int main()
{
	int i,k;
	while(scanf("%d",&n),n)
	{
		m=n*(n-1)/2;
		int a,b,c,e;
		
		into();   //初始化根节点 

		k=0;
		for(i=0;i<m;++i)
		{
			scanf("%d%d%d%d",&a,&b,&c,&e);
			if(e==1)  //已有路，不用再修，将他们连通看做 一个整体 
				join(a,b);
			else if(e==0)   //没连通，用结构体保存 
			{
				s[k].u=a;
				s[k].v=b;
				s[k].w=c;
				k++;
			}
		}
		
		sort(s,s+k,cmp);   //按成本从小到大排序 
		
		int sum=0;
		for(i=0;i<k;++i)
		{
			if(join(s[i].u,s[i].v)) //计算成本 
				sum+=s[i].w;
		}
		printf("%d\n",sum);
	}
	return 0;
}

已AC代码：（prim 思想）已修好的赋值为 0

#include<cstdio>
#include<cstring>
#define INF 0xfffffff

int map[125][125],low[125],vis[125];
int n,m,sum;  //map二维数组存图，low记录每2个点间最小权值，vis标记某点是否已访问

void prim()
{
	memset(vis,0,sizeof(vis));
	int i,j,min,pos;
	for(i=1;i<=n;++i)  //从某点开始，分别标记vis和记录该点pos
		low[i]=map[1][i];  //第一次给low数组赋值 map的第一行 
	vis[1]=1;	pos=1;
	for(i=1;i<n;++i)   //再运行n-1次,一次找一个最小 
	{
		min=INF;
		for(j=1;j<=n;++j)
		{
			if(vis[j]==0&&low[j]<min)
			{
				min=low[j];
				pos=j;
			}
		}

		vis[pos]=1;   //标记该点已访问 
		sum+=min;   //最小权值累加
		
		for(j=1;j<=n;++j)    //更新权值low 把 map的 pos 行中比对应的 low 小的赋给low 
			if(vis[j]==0&&low[j]>map[pos][j])
				low[j]=map[pos][j];
	}
	return ;
}

int main()
{
	int a,b,c,d,i,j;
	while(scanf("%d",&n),n)
	{
		m=n*(n-1)/2;
		for(i=1;i<=n;++i)  //没路的 赋为无穷大，不再考虑 
			for(j=1;j<=n;++j)
				map[i][j]=INF;
		while(m--)
		{
			scanf("%d%d%d%d",&a,&b,&c,&d);
			if(d==0)
			{
				map[a][b]=map[b][a]=c;
			}
			else if(d==1)  //已连通 赋为 0 代表 两村子距离为 0 
				map[a][b]=map[b][a]=0;
		}
		
//		printf("\n");
//		for(i=1;i<=n;++i) 
//		{
//			for(j=1;j<=n;++j)
//			{
//				printf("%10d  ",map[i][j]);
//			}
//			printf("\n");
//		}
//		printf("\n");
			
		
		sum=0;
		prim();
		printf("%d\n",sum);
	}
	return 0;
}