HDU-1879-继续畅通工程

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1879

继续畅通工程

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
  
  

 

prim算法  与模板的区别 当f为0的时候，不用修路，在开始读入边的时候，让他的费用为0. （用cin会超时）

代码如下：

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int n,u,v,w,f;
int e[111][111],visit[111],dis[111];
const int inf=999999;
void init()
{
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			if(i==j)
			e[i][j]=0;
			else
			e[i][j]=inf;
		}
	}
}
void prim()
{
	int i,j,k,minn;
	for(i=1;i<=n;i++)
	{
		dis[i]=e[1][i];
	}
	memset(visit,0,sizeof(visit));
	visit[1]=1;
	int sum=0,count=1;
	while(count<n)
	{
		minn=inf;
		for(i=1;i<=n;i++)
		{
			if(visit[i]==0&&dis[i]<minn)
			{
				minn=dis[i];
				j=i;
			}
		}
		visit[j]=1;
		count++;
		sum=sum+dis[j];
		for(k=1;k<=n;k++)
		{
			if(visit[k]==0&&dis[k]>e[j][k])
			dis[k]=e[j][k];
		}
	}
	printf("%d\n",sum);
}
int main()
{
	while(scanf("%d",&n))
	{
		if(n==0)
		break;
		int m=n*(n-1)/2;
		init();
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d%d",&u,&v,&w,&f);//这道题用cin会超时
			if(f==0)
			{
				e[u][v]=w;
			    e[v][u]=w;
			}
			else//让已连通的费用为0 
			{
				e[u][v]=0;
			    e[v][u]=0;
			}
		}
		prim();
	}
	return 0;
 }

kruskal 算法

代码如下：

#include<iostream>  
#include<cstdio>  
#include<cstring>  
#include<algorithm>  
using namespace std;  
struct edge{  
    int u,v,w;  
}e[11111];  
int n,f;
int pre[11111];  
int find(int v)  
{  
    if(pre[v]==v)  
    return v;  
    else  
    {  
        pre[v]=find(pre[v]);  
        return pre[v];  
    }  
 }   
int merge(int u,int v)  
{  
    int t1=find(u);  
    int t2=find(v);  
    if(t1!=t2)   
    {  
        pre[t2]=t1;  
        return 1;   
    }  
    return 0;  
 }   
 bool cmp(edge a,edge b)  
 {  
    return a.w<b.w;  
  }   
int main()  
{  
    int i;  
    while(~scanf("%d",&n))
    {
        int m=n*(n-1)/2;
        if(n==0)
        break;
        for(i=1;i<=m;i++)  
        {  
            scanf("%d%d%d%d",&e[i].u,&e[i].v,&e[i].w,&f);
            if(f)
            e[i].w=0; 
        }  
        sort(e+1,e+m+1,cmp);  
        //初始化   
        for(i=1;i<=n;i++)  
        {  
            pre[i]=i;  
        }  
        int count=0,sum=0;  
        for(i=1;i<=m;i++) 
        {   
            if(merge(e[i].u,e[i].v))  
            {  
                count++;  
                sum=sum+e[i].w;  
             }  
            if(count==n-1)   
            break;  
         }   
        printf("%d\n",sum);        
    }   
    return 0;  
}

或者 把给出能连通的点 让他们的根节点相同，但是初始化要在前面。

#include<iostream>  
#include<cstdio>  
#include<cstring>  
#include<algorithm>  
using namespace std;  
struct edge{  
    int u,v,w;  
}e[11111];  
int n,f;
int pre[11111];  
int find(int v)  
{  
    if(pre[v]==v)  
    return v;  
    else  
    {  
        pre[v]=find(pre[v]);  
        return pre[v];  
    }  
 }   
int merge(int u,int v)  
{  
    int t1=find(u);  
    int t2=find(v);  
    if(t1!=t2)   
    {  
        pre[t2]=t1;  
        return 1;   
    }  
    return 0;  
 }   
 bool cmp(edge a,edge b)  
 {  
    return a.w<b.w;  
  }   
int main()  
{  
    int i;  
    while(~scanf("%d",&n))
    {
        int m=n*(n-1)/2;
        if(n==0)
        break;
        for(i=1;i<=n;i++)  
        {  
            pre[i]=i;  
        }  
        for(i=1;i<=m;i++)  
        {  
            scanf("%d%d%d%d",&e[i].u,&e[i].v,&e[i].w,&f);
            if(f)
            {
            	int fx=find(e[i].u);
            	int fy=find(e[i].v);
            	if(fx!=fy)
            	{
            		pre[fy]=fx;
				}
			}
        }  
        sort(e+1,e+m+1,cmp);  
       
        int count=0,sum=0;  
        for(i=1;i<=m;i++) 
        {   
            if(merge(e[i].u,e[i].v))  
            {  
                count++;  
                sum=sum+e[i].w;  
             }  
            if(count==n-1)   
            break;  
         }   
        printf("%d\n",sum);        
    }   
    return 0;  
}