【洛谷P1433】吃奶酪【状压DP】

题目描述

房间里放着 n 块奶酪。一只小老鼠要把它们都吃掉，问至少要跑多少距离？老鼠一开始在 (0,0) 点处。

输入格式

第一行有一个整数，表示奶酪的数量 n。

第 2 到第 (n + 1) 行，每行两个实数，第 (i + 1) 行的实数分别表示第 i 块奶酪的横纵坐标 x_i, y_i

输出格式

输出一行一个实数，表示要跑的最少距离，保留 2 位小数。

输入输出样例

输入 #1

4
1 1
1 -1
-1 1
-1 -1

输出 #1

7.41

数据规模与约定

对于全部的测试点，保证 n<=15

分析

看数据n<=15就知道状压DP。

然后设$f[j][i]$为第j个点的状态为i，枚举$i,j,k$。状态转移方程：
$$f[j][i]=min(f[j][i],f[k][i-(1<<j-1)]+dis(k,j))$$
dis函数就是算距离，所以所有数组开double。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iomanip> 
typedef long long ll;
using namespace std;

double x[201],y[201],f[20][1<<15];
int n;

double dis(int l,int r)
{
	return sqrt((x[l]-x[r])*(x[l]-x[r])+(y[l]-y[r])*(y[l]-y[r]));
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
    	cin>>x[i]>>y[i];
    }
    for(int i=1;i<=n;i++)
    {
    	for(int j=1;j<=(1<<n)-1;j++)
    	{
    		f[i][j]=2147483647;
    	}
    }
    for(int i=0;i<=(1<<n)-1;i++)
    {
    	for(int j=1;j<=n;j++)
    	{
    		if(!(i&(1<<j-1))) continue;
    		if(i==(1<<j-1))
    		{
    			f[j][i]=0;
    			continue;
    		}
    		for(int k=1;k<=n;k++)
    		{
    			if(j==k||(i&(1<<k-1))==0)
    			{
    				continue;
    			}
    			f[j][i]=min(f[j][i],f[k][i-(1<<j-1)]+dis(k,j));
    		}
    	}
    }
    double ans=2147483647;
    for(int i=1;i<=n;i++)
    {
    	ans=min(ans,f[i][(1<<n)-1]+dis(i,0));
    }
    cout<<setprecision(2)<<fixed<<ans;
	return 0;
}