白书P61 - 点集配对问题

白书P61 - 点集配对问题

状压DP

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define INF 0x3f3f3f3f

struct Point
{
    double x,y,z;
}p[+];

int n;
double dp[(<<)+];            //dp[j]表示j对应状态（0为未配对，1为配对了）的最小距离和

double dist(int i,int j)
{
    return sqrt((p[i].x-p[j].x)*(p[i].x-p[j].x)+(p[i].y-p[j].y)*(p[i].y-p[j].y)+(p[i].z-p[j].z)*(p[i].z-p[j].z));
}

void solve()
{
    int i,j,k,MAX=<<n;
    for(i=;i<MAX;i++)
    {
        dp[i]=1e10;
    }
    dp[]=;

    for(j=;j<MAX;j++)
    {
        for(i=;i<n;i++)
        {
            if(j&(<<i)) break;
        }
        for(k=i+;k<n;k++)
        {
            if(j&(<<k)
            {
                dp[j]=min(dp[j],dp[(j&(~(<<i)))&(~(<<k))]+dist(i,k));
            }
        }
    }
    printf("%lf\n",dp[MAX-]);
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=;i<n;i++)
        {
            scanf("%lf%lf%lf",&p[i].x,&p[i].y,&p[i].z);
        }
        solve();
    }
    return ;
}