5-3 畅通工程之局部最小花费问题 (35分)

5-3 畅通工程之局部最小花费问题   (35分)

某地区经过对城镇交通状况的调查，得到现有城镇间快速道路的统计数据，并提出“畅通工程”的目标：使整个地区任何两个城镇间都可以实现快速交通（但不一定有直接的快速道路相连，只要互相间接通过快速路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建快速路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全地区畅通需要的最低成本。

输入格式:

输入的第一行给出村庄数目$N$ ($1\le N\le 100$)；随后的$N(N-1)/2$行对应村庄间道路的成本及修建状态：每行给出4个正整数，分别是两个村庄的编号（从1编号到$N$），此两村庄间道路的成本，以及修建状态 — 1表示已建，0表示未建。

输出格式:

输出全省畅通需要的最低成本。

输入样例:

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

输出样例:

3

#include<stdio.h>
#include<algorithm>
using namespace std;
int  G[1002][1002];
int d[1002];
bool vis[1003]={false};
int inf=0xfffffff;
int ans,n;
int a()
{
	d[1]=0;
	ans=0;
	for(int i=1;i<=n;i++)
	{
		int u=-1,min=inf;
		for(int j=1;j<=n;j++)
		{
			if(vis[j]==false&&d[j]<min)
			{
				u=j;
				min=d[j];
			}
		}
		vis[u]=true;
		ans+=d[u];
		for(int v=1;v<=n;v++)
		{
			if(vis[v]==false&&G[u][v]!=inf&&G[u][v]<d[v])
			{
				d[v]=G[u][v];
			}
		}
	}
	return ans;
}//以上为最小生成树模板 
int main()
{
	int u,v,w,x;
    scanf("%d",&n);
		for(int i=0;i<1002;i++)
	{
		d[i]=inf;
		vis[i]=false;
		for(int j=0;j<1002;j++)
		{
		     G[i][j]=inf;
		}
	}
	int m=((n-1)*n)/2;
	while(m--)
	{
		scanf("%d%d%d%d",&u,&v,&w,&x);
		if(x==0)
		{
		G[u][v]=w;G[v][u]=w;
	   }
	   else
	   {
	   	G[u][v]=0;G[v][u]=0;
	   }//以上二个判断是两庄的成本 
}
  int ans=a();
  printf("%d\n",ans);
  return 0;
 }