hdu2489 Minimal Ratio Tree

hdu2489 Minimal Ratio Tree

题意：一个 至多  n=15 的 完全图 ，求 含有 m 个节点的树 使 边权和 除 点权和 最小

题解：枚举 m 个 点 ，然后 求 最小生成树

自己粗心。。。。WA 了 好多次……（233333 ）

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <queue>
#include <stack>
#include <set>
#include <string>
using namespace std;
typedef long long ll;
const double ESP = 10e-8;
const int MOD = 1000000007;
typedef long long LL;
const int MAXN = 15 + 3;

int graph[MAXN][MAXN];
int node[MAXN];
int tmp[MAXN];
int dist[MAXN];
int ans[MAXN];
double ansMi;
int n,m;
bool vis[MAXN];
double prim(){
    double dis = 0;
    memset(vis,0,sizeof(vis));
    int cur = 0;
    vis[cur] = 1;
    for(int i = 0;i < m;i++){
        dist[i] = graph[ tmp[cur] ][ tmp[i] ];
    }

    for(int i = 0;i < m-1;i++){
        int mi = 0x7ffffff;
        int k;
        for(int j = 0;j < m;j++){
            if(!vis[j] && dist[j] < mi){
                mi = dist[j];
                k = j;
            }
        }
        cur = k;
        vis[cur] = 1;
        dis += mi;
        for(int j = 0;j < m;j++){
            if(!vis[j] && dist[j] > graph[ tmp[cur] ][ tmp[j] ]){
                dist[j] = graph[ tmp[cur] ][ tmp[j] ];
            }
        }
    }
    return dis;
}

void dfs(int v,int cnt){
    if(cnt == m-1){
        double h = 0;
        for(int i = 0;i < m;i++){
            h += node[ tmp[i] ];
        }
        double b = prim();
        double tt = b/h;
        if(tt - ansMi < -(1e-9)){
            ansMi = tt;
            for(int i = 0;i < m;i++){
                ans[i] = tmp[i];
            }
        }
        return;
    }
    for(int i = v+1;i<= n;i++){
        tmp[cnt+1] = i;
        dfs(i,cnt+1);
    }
}

int main(){
//    freopen("input.txt","r",stdin);
    while(~scanf("%d%d",&n,&m) && n &&m){
        for(int i = 1;i <= n;i++){
            scanf("%d",&node[i]);
        }
        for(int i = 1;i <= n;i++){
            for(int j = 1;j <= n;j++){
                scanf("%d",&graph[i][j]);
            }
        }
        ansMi = 10e10;
        for(int i = 1;i <= n;i++){
            tmp[0] = i;
            dfs(i,0);
        }
        for(int i = 0;i < m-1;i++){
            printf("%d ",ans[i]);
        }
        printf("%d
",ans[m-1]);
    }
    return 0;
}