BZOJ 1097 POI2007 旅游景点atr SPFA+状压DP

题目大意：给定一张图，要求从第一个点出发，按照某个拓扑序遍历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;
}