[BZOJ1097][POI2007]旅游景点atr（状压dp）

题目描述

传送门

题解

这道题的题目描述比较奇怪，而且觉得A掉了之后写的还有小问题= =
不过最大的收获是pre数组的使用：判断之前要经过哪些点。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;

const int INF=1e9;
const int max_n=2e4+5;
const int max_m=2e5+5;
const int max_e=max_m*2;
const int N=20;

int n,m,k,tot,head,tail,ans=INF;
int cnt,point[max_n],next[max_e],v[max_e],c[max_e];
int pre[max_n],dis[N+5][N+5],f[1<<N][N+5],d[max_n],len[max_n];
bool vis[max_n];
int q[max_m];

inline void add(int x,int y,int z){++cnt,next[cnt]=point[x],point[x]=cnt,v[cnt]=y,c[cnt]=z;}
inline void spfa(int s){
    memset(q,0,sizeof(q));
    head=tail=0; q[++tail]=s;
    memset(vis,0,sizeof(vis)); memset(d,0x7f,sizeof(d)); d[s]=0;
    while (head!=tail){
        int now=q[++head]; 
        vis[now]=false;
        for (int i=point[now];i;i=next[i])
          if (d[v[i]]>d[now]+c[i]){
            d[v[i]]=d[now]+c[i];
            if (!vis[v[i]]) vis[v[i]]=true,q[++tail]=v[i];
          }
    }
    for (int i=1;i<=k+1;++i)
      dis[s][i]=d[i];
    len[s]=d[n];
}
int main(){
    scanf("%d%d%d",&n,&m,&k);
    int x,y,z;
    for (int i=1;i<=m;++i) 
      scanf("%d%d%d",&x,&y,&z),add(x,y,z),add(y,x,z);
    scanf("%d",&m);
    for (int i=1;i<=m;++i) 
      scanf("%d%d",&x,&y),pre[y]|=1<<(x-2);
    for (int i=1;i<=k+1;++i)
      spfa(i);
    tot=(1<<k)-1;
    memset(f,-1,sizeof(f));
    f[0][1]=0;
    for (int p=0;p<=tot;p++)
      for (int i=1;i<=k+1;i++)
        if (f[p][i]!=-1)
          for (int j=2;j<=k+1;j++){
             int nxt=p|(1<<(j-2));
             if ((p&pre[j])==pre[j])
              if (f[nxt][j]==-1||f[nxt][j]>f[p][i]+dis[i][j])
               f[nxt][j]=f[p][i]+dis[i][j];
          }
    for (int i=1;i<=k+1;++i)
      if (f[tot][i]!=-1)
        ans=min(ans,f[tot][i]+len[i]);
    printf("%d\n",ans);
}