题目大意:给定一张图,要求从第一个点出发,按照某个拓扑序遍历2~k+1的所有节点,然后到达n,求最短路径
首先将所有关键点之间的最短路用SPFA求出来 然后状压DP
令f[state][p]表示已经走过的点集为state,将要走到p点的最短路
记忆化搜索就行了- -
标准卡时过- -
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 20200
using namespace std;
struct abcd{
int to,f,next;
}table[200200<<1];
int head[M],tot;
int n,m,k;
int map[30][30],topology_map[23];
int f[1<<21][23];
void Add(int x,int y,int z)
{
table[++tot].to=y;
table[tot].f=z;
table[tot].next=head[x];
head[x]=tot;
}
void SPFA(int s)
{
static int q[65540],f[M];
static unsigned short r,h;
static bool v[M];
int i;
memset(f,0x3f,sizeof f);
q[++r]=s;f[s]=0;
while(r!=h)
{
int x=q[++h];v[x]=0;
for(i=head[x];i;i=table[i].next)
if(f[table[i].to]>f[x]+table[i].f)
{
f[table[i].to]=f[x]+table[i].f;
if(!v[table[i].to])
v[table[i].to]=1,q[++r]=table[i].to;
}
}
for(i=1;i<=k+1;i++)
map[i][s<=k+1?s:k+2]=f[i];
}
int State_Compressed_DP(int state,int p)
{
if(f[state][p])
return f[state][p];
if(!state)
return 0;
int i;
f[state][p]=0x3f3f3f3f;
for(i=1;i<=k+1;i++)
if( state&(1<<i-1) && (topology_map[i]&state)==0 )
f[state][p]=min(f[state][p],State_Compressed_DP(state^(1<<i-1),i)+map[i][p]);
return f[state][p];
}
int main()
{
//freopen("atr11b.in","r",stdin);
int i,x,y,z,temp;
cin>>n>>m>>k;
for(i=1;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&z);
Add(x,y,z);Add(y,x,z);
}
cin>>temp;
for(i=1;i<=temp;i++)
{
scanf("%d%d",&x,&y);
topology_map[x]|=1<<y-1;
}
for(i=2;i<=k+1;i++)
topology_map[1]|=1<<i-1;
for(i=1;i<=k+1;i++)
SPFA(i);SPFA(n);
cout<<State_Compressed_DP((1<<k+1)-1,k+2)<<endl;
return 0;
}