继续畅通工程
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时输入结束。
当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
思路: 将已连通的看做一个整体,将没连通的存入结构体,根据 克鲁斯卡尔 思想,求最低成本
例如:
第二组测试数据数据:
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;
}