HDU-1879-继续畅通工程

来源:互联网 发布:电脑网络有时有有时没 编辑:程序博客网 时间:2024/06/03 17:03

题目链接: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
31 2 1 01 3 2 02 3 4 031 2 1 01 3 2 02 3 4 131 2 1 01 3 2 12 3 4 10
 

Sample Output
310
 
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;elsee[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;  }


原创粉丝点击