pku hdu Minimal Ratio Tree

枚举最小生成树

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int map[20][20];
int node[20],n,m,sol[20];
double best;
bool ok(int x)
{
    int ans=0;
    while(x)
    {
        ans+=x&1;
        x>>=1;
    }
    return ans==m;
}
void prim(int x)
{
    bool v[20]= {0};
    int dis[20],cursol[20],s;
    double se=0,sp=0,cur;
    for(s=0; s<n; s++)
        if(x&(1<<s)) break;
    for(int i=0; i<n; i++)
        dis[i]=map[s][i];
    int num=0;
    for(int i=0;i<n;i++)
        if(x&(1<<i)) sp+=node[i],cursol[num++]=i;

    v[s]=1;
    while(1)
    {
        int mmin=1<<30,u=-1;
        for(int i=0;i<n;i++)
        {
            if((x&(1<<i))==0) continue;
            if(!v[i]&&dis[i]<mmin) u=i,mmin=dis[i];
        }
        if(u==-1) break;
        v[u]=1,se+=mmin;
        for(int i=0; i<n; i++)
            if(map[u][i]<dis[i]&&!v[i]) dis[i]=map[u][i];
    }
    cur=se/sp;
    if(cur==best)
    {
        bool f=false;
        for(int i=0;i<m&&!f;i++)
            {
                if(sol[i]>cursol[i]) f=true;
                if(sol[i]<cursol[i]) break;
            }
        if(f)
        {
            for(int i=0;i<m;i++)
                sol[i]=cursol[i];
        }
    }
    else if(cur<best)
    {
        best=cur;
        for(int i=0;i<m;i++)
            sol[i]=cursol[i];
    }
}
int main()
{

    while(scanf("%d%d",&n,&m))
    {
        if(n==0&&m==0) break;
        for(int i=0;i<n;i++) scanf("%d",&node[i]);
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&map[i][j]);
        int cnt=1<<n;
        best=1e50,sol[0]=0;
        for(int i=0;i<cnt;i++) if(ok(i)) prim(i);
        for(int i=0;i<m-1;i++)
            printf("%d ",sol[i]+1);
        printf("%d/n",sol[m-1]+1);
    }
    return 0;
}