POJ3411 状态压缩DP
题目链接
链接: http://poj.org/problem?id=3411.
题目大意
给N个城市,m条路,并给出m条路的信息,每条信息给出起点a终点b和中间节点c,有两种付款方式
1、在中间节点c付款(就是可以提前交钱)-R(a,b);
2、到终点b付款-p(a,b)。
问1到N的最小花费
思路
dp[i][S] 经过在集合s中的点到达i的最短花费
1、当S中没有c时
dp[b][s∪b]=min(dp[i][s],dp[a][s]+P(a,b));
2、当S中有c时;
dp[b][s∪b]=min(dp[i][s],dp[a][s]+R(a,b),dp[a][s]+P(a,b));
先将路径按起点城市的序号升序排序;
首先如果一条路径缩短,在最短花费的路上肯定不会经过终点;
在第一次遍历中,可以求出城市号升序经过的最小花费路径;
每一次外层循环其实都是因为有回路缩短了最小距离,因为回路就会产生中间节点c,就可以使用第一种付费方案
每一次循环的最短回路只会被更新一次,比如n1到n2到n1是一条回路,缩短了n1到n3的值,因为是正序遍历,所以n3到n1的回路值后跟新,但是下次外循环时不会再缩短n1到n2到n1到n3这条回路而是缩短n1到n3到n1到ni>3的回路,所以最多需要N-1次外层循环!
AC代码
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
int dp[15][1<<11]; //到i点时经过了集合s
struct edge
{
int a,b,c,p,r;
bool operator<(const edge B)const
{
if(a==B.a&&b==B.b)
return c<B.c;
if(a==B.a)
return b<B.b;
return a<B.a;
}
}arr[15];
int N,m;
int main()
{
while(scanf("%d%d",&N,&m)!=EOF)
{
int i,s;
for(i=1;i<=m;i++)
scanf("%d%d%d%d%d",&arr[i].a,&arr[i].b,&arr[i].c,&arr[i].p,&arr[i].r);
memset(dp,INF,sizeof(dp));
dp[1][1]=0;
sort(arr+1,arr+m+1);
for(int j=1;j<N;j++) //!!!
for(i=1;i<=m;i++)
{
for(s=1;s<(1<<N);s++)
{
if(!(s&(1<<(arr[i].a-1)))||dp[arr[i].a][s]==INF)continue;
dp[arr[i].b][s|(1<<(arr[i].b-1))]=min(dp[arr[i].b][s|(1<<(arr[i].b-1))],dp[arr[i].a][s]+arr[i].r);
if(s&(1<<(arr[i].c-1)))
dp[arr[i].b][s|(1<<(arr[i].b-1))]=min(dp[arr[i].b][s|(1<<(arr[i].b-1))],dp[arr[i].a][s]+arr[i].p);
}
}
int ans=INF;
for(s=1;s<(1<<N);s++)
ans=min(ans,dp[N][s]);
if(ans==INF)
printf("impossible\n");
else
printf("%d\n",ans);
}
return 0;
}