HDOJ 1863 prim算法 HDOJ 1879

 #include<cstdio>
 #include<cstring>
 #define inf 0xffffff
 int g[][];
 int ans;
 void prim(int n)
 {
     int lowcost[],used[],i,j,k,min,closet[];
     memset(used,,sizeof(used));
     for(i=;i<=n;i++)
         lowcost[i]=g[i][],closet[i]=;
     used[]=;
     for(i=;i<n;i++)
     {
         j=;
         min=inf;
         for(k=;k<=n;k++)
         {
             if(lowcost[k]<min&&!used[k])
                 min=lowcost[k],
                 j=k;
         }
         ans+=g[j][closet[j]];
         used[j]=;
         for(k=;k<=n;k++)
         {
             if(g[k][j]<lowcost[k]&&!used[k])
                 lowcost[k]=g[k][j],
                 closet[k]=j;
         }
     }

 }
 int main()
 {
     int m,n;
     while(scanf("%d %d",&m,&n)!=EOF&&m)
     {
         int i,j;
         ans=;
         for(i=;i<=n;i++)
             for(j=;j<=n;j++)
                 g[i][j]=inf;
         int a,b,t;
         for(i=;i<=m;i++)
         {
             scanf("%d %d %d",&a,&b,&t);
             if(g[a][b]>t)
                 g[a][b]=g[b][a]=t;
         }
         if(m<n-)
             printf("?\n");
         else
         {
            prim(n);
            if(ans>=inf)
                printf("?\n");
            else
                printf("%d\n",ans);
         }
     }

     return ;
 }

9896382

2013-12-26 16:56:48

Accepted

1863

0MS

296K

989 B

C++

泽泽

1879

9896417

2013-12-26 17:03:42

Accepted

1879

234MS

300K

934 B

C++

泽泽

 #include<cstdio>
 #include<cstring>
 #define inf 0xffffff
 int g[][];
 int ans;
 void prim(int n)
 {
     int lowcost[],used[],i,j,k,min,closet[];
     memset(used,,sizeof(used));
     for(i=;i<=n;i++)
         lowcost[i]=g[i][],closet[i]=;
     used[]=;
     for(i=;i<n;i++)
     {
         j=;
         min=inf;
         for(k=;k<=n;k++)
         {
             if(lowcost[k]<min&&!used[k])
                 min=lowcost[k],
                 j=k;
         }
         ans+=g[j][closet[j]];
         used[j]=;
         for(k=;k<=n;k++)
         {
             if(g[k][j]<lowcost[k]&&!used[k])
                 lowcost[k]=g[k][j],
                 closet[k]=j;
         }
     }

 }
 int main()
 {
     int n;
     while(scanf("%d",&n)!=EOF&&n)
     {
         int i,j;
         ans=;
         for(i=;i<=n;i++)
             for(j=;j<=n;j++)
                 g[i][j]=inf;
         int a,b,t;
         for(i=;i<=n*(n-)/;i++)
         {
             scanf("%d %d %d %d",&a,&b,&t,&j);
             if(g[a][b]>t&&j==)
                 g[a][b]=g[b][a]=t;
             else if(j==)
                 g[a][b]=g[b][a]=;
         }
         prim(n);
         printf("%d\n",ans);
     }

     return ;
 }