白书P61 - 点集配对问题

白书P61 - 点集配对问题

状压DP

#include <iostream>

#include <cstdio>

#include <cstring>

using namespace std;

#define INF 0x3f3f3f3f



struct Point

{

    double x,y,z;

}p[20+2];



int n;

double dp[(1<<20)+10];            //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=1<<n;

    for(i=0;i<MAX;i++)

    {

        dp[i]=1e10;

    }

    dp[0]=0;



    for(j=0;j<MAX;j++)

    {

        for(i=0;i<n;i++)

        {

            if(j&(1<<i)) break;

        }

        for(k=i+1;k<n;k++)

        {

            if(j&(1<<k)

            {

                dp[j]=min(dp[j],dp[(j&(~(1<<i)))&(~(1<<k))]+dist(i,k));

            }

        }

    }

    printf("%lf\n",dp[MAX-1]);

}

int main()

{

    while(scanf("%d",&n)!=EOF)

    {

        for(int i=0;i<n;i++)

        {

            scanf("%lf%lf%lf",&p[i].x,&p[i].y,&p[i].z);

        }

        solve();

    }

    return 0;

}