hdu 2489 Minimal Ratio Tree

http://acm.hdu.edu.cn/showproblem.php?pid=2489

这道题就是n个点中选择m个点形成一个生成树使得生成树的ratio最小。暴力枚举+最小生成树。

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #define maxn 1000
 using namespace std;
 const int inf=<<;
 const double eps=1e-;

 int map[maxn][maxn];
 int g[maxn][maxn];
 bool vis[maxn];
 int dis[maxn];
 int node[maxn];
 int ok[maxn];
 int w[maxn];
 int n,m;
 double ans;

 int prime()
 {
     memset(vis,false,sizeof(vis));
     for(int i=; i<=m; i++) dis[i]=g[][i];
     dis[]=;
     vis[]=true;
     int sum1=;
     for(int i=; i<m; i++)
     {
         int x,m2=inf;
         for(int y=; y<=m; y++) if(!vis[y]&&dis[y]<m2) m2=dis[x=y];
         vis[x]=true;
         sum1+=m2;
         for(int y=; y<=m; y++) if(!vis[y]&&dis[y]>g[x][y]){
             dis[y]=g[x][y];
         }
     }
     return sum1;
 }

 void dfs(int u,int cnt)
 {
     node[cnt]=u;
     if(cnt>=m)
     {
         for(int i=; i<=m; i++)
         {
             for(int j=; j<=m; j++)
             {
                 g[i][j]=map[node[i]][node[j]];
             }
         }
         double edgew=prime()*1.0;
         double nodew=0.0;
         for(int i=; i<=m; i++)
         {
             nodew+=w[node[i]];
         }
         double sum=edgew/nodew;
         if(ans-sum>=eps)
         {
             ans=sum;
             for(int i=; i<=m; i++)
              ok[i]=node[i];
         }
     }
     for(int j=u+; j<=n; j++)
     {
         dfs(j,cnt+);
     }
 }

 int main()
 {
     while(scanf("%d%d",&n,&m)!=EOF)
     {
         memset(node,,sizeof(node));
         memset(map,,sizeof(map));
         memset(w,,sizeof(w));
         memset(ok,,sizeof(ok));
         if(n==&&m==) break;
         for(int i=; i<=n; i++)
             scanf("%d",&w[i]);
         for(int i=; i<=n; i++)
         {
             for(int j=; j<=n; j++)
             {
                 scanf("%d",&map[i][j]);
             }
         }
         ans=inf;
         for(int i=; i<=n; i++)
         {
             dfs(i,);
         }
         for(int j=; j<=m; j++)
         {
             if(j==) printf("%d",ok[j]);
             else printf(" %d",ok[j]);
         }
         printf("\n");
     }
     return ;
 }