题目描述
某乡有n个村庄(1
输入输出格式
输入格式:
村庄数n和各村之间的路程(均是整数)。
输出格式:
最短的路程。
输入输出样例
输入样例#1:
3
0 2 1
1 0 2
2 1 0
输出样例#1:
3
说明
输入解释
3 {村庄数}
0 2 1 {村庄1到各村的路程}
1 0 2 {村庄2到各村的路程}
2 1 0 {村庄3到各村的路程}
思路
看到20想到状压。
这里
DP[i][j]
表示i情况下走到j。
然后直接转移。
如果可以转移,转移方程是:
dp[k][i|bit[k]]=min(dp[k][i|bit[k]],dp[j][i]+map[j][k]);
但是不知道为什么最后的一个点TLE了。。
不过也是啊,
520000∗20∗20
会挂啊。。。
我还以为洛谷跟香港记者一样呢
代码
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
const int N=20+1;
const int inf=0x3f3f3f3f;
int bit[N],map[N][N],dp[N][1<<20],n,ans=inf,tot;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') {x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int main()
{
n=read();
bit[1]=1;
memset(dp,inf,sizeof(dp));
for (int i=2;i<=n;i++)
bit[i]=bit[i-1]*2;
tot=(1<<n)-1;
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
map[i][j]=read();
dp[1][1]=0;
for (int i=1;i<=tot;i+=2)
for (int j=1;j<=n;j++)
if (dp[j][i]<=inf)
{
for (int k=1;k<=n;k++)
if (!(i&bit[k])) dp[k][i|bit[k]]=min(dp[k][i|bit[k]],dp[j][i]+map[j][k]);
}
for (int i=1;i<=n;i++)
ans=min(ans,dp[i][tot]+map[i][1]);
printf("%d\n",ans);
return 0;
}
/*
3
0 2 1
1 0 2
2 1 0
*/