POJ3411 状态压缩DP

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，就可以使用第一种付费方案
每一次循环的最短回路只会被更新一次，比如n₁到n₂到n₁是一条回路，缩短了n₁到n₃的值，因为是正序遍历，所以n₃到n₁的回路值后跟新，但是下次外循环时不会再缩短n₁到n₂到n₁到n₃这条回路而是缩短n₁到n₃到n₁到n_i>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;
}