bzoj 1097: [POI2007]旅游景点atr（状压DP）

1097: [POI2007]旅游景点atr

Time Limit: 30 Sec   Memory Limit: 357 MB
Submit: 1509   Solved: 343
[ Submit][ Status][ Discuss]

Description

　　FGD想从成都去上海旅游。在旅途中他希望经过一些城市并在那里欣赏风景，品尝风味小吃或者做其他的有趣
的事情。经过这些城市的顺序不是完全随意的，比如说FGD不希望在刚吃过一顿大餐之后立刻去下一个城市登山，
而是希望去另外什么地方喝下午茶。幸运的是,FGD的旅程不是既定的，他可以在某些旅行方案之间进行选择。由于
FGD非常讨厌乘车的颠簸，他希望在满足他的要求的情况下，旅行的距离尽量短，这样他就有足够的精力来欣赏风
景或者是泡MM了^_^.整个城市交通网络包含N个城市以及城市与城市之间的双向道路M条。城市自1至N依次编号，道
路亦然。没有从某个城市直接到它自己的道路，两个城市之间最多只有一条道路直接相连，但可以有多条连接两个
城市的路径。任意两条道路如果相遇，则相遇点也必然是这N个城市之一，在中途，由于修建了立交桥和下穿隧道
，道路是不会相交的。每条道路都有一个固定长度。在中途，FGD想要经过K(K<=N-2)个城市。成都编号为1，上海
编号为N,而FGD想要经过的N个城市编号依次为2,3,…,K+1.举例来说，假设交通网络如下图。FGD想要经过城市2,3,
4,5，并且在2停留的时候在3之前，而在4,5停留的时候在3之后。那么最短的旅行方案是1-2-4-3-4-5-8,总长度为1
9。注意FGD为了从城市2到城市4可以路过城市3,但不在城市3停留。这样就不违反FGD的要求了。并且由于FGD想要
走最短的路径，因此这个方案正是FGD需要的。

Input

　　第一行包含3个整数N(2<=N<=20000),M(1<=M<=200000),K(0<=K<=20),意义如上所述。

Output

　　只包含一行，包含一个整数，表示最短的旅行距离。

Sample Input

8 15 4
1 2 3
1 3 4
1 4 4
1 6 2
1 7 3
2 3 6
2 4 2
2 5 2
3 4 3
3 6 3
3 8 6
4 5 2
4 8 6
5 7 4
5 8 6
3
2 3
3 4
3 5

Sample Output

19

HINT

 上面对应于题目中给出的例子。

Source

题解：状压DP

T_T调了好久发现是数组开小了，撞死的心都有了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define N 20
#define M 400003
#define inf 1e9
using namespace std;
int n,m,k,t;
int dis[N*2][N*2],d[M],can[M],len[M],a[N*2];
int f[1048576][22],mi[N*2],tal;
int v[M],point[M],next[M],u[M],tot;
void add(int x,int y,int z)
{
   tot++; next[tot]=point[x]; u[tot]=y; v[tot]=z; point[x]=tot;
   tot++; next[tot]=point[y]; u[tot]=x; v[tot]=z; point[y]=tot; 
}
void spfa(int x)//处理出必须要经过的点之间的最短路
{
    for (int i=1;i<=n;i++)
     d[i]=inf;
    memset(can,0,sizeof(can));
    d[x]=0; can[x]=1;
    queue<int> p;
    p.push(x);
    while (!p.empty())
    {
        int now=p.front(); p.pop();
        for (int i=point[now];i;i=next[i])
         if (d[now]+v[i]<d[u[i]])
          {
            d[u[i]]=d[now]+v[i];
            if (!can[u[i]])
             {
                can[u[i]]=1;
                p.push(u[i]);
             }
          }
        can[now]=0;
    }
    len[x]=d[n];//记录当前点到N的路径，因为无论怎么走最后都要到N
    for (int i=1;i<=k+1;i++)
     if (i!=x)
      dis[x][i]=d[i];
}
void dp()
{
	for (int p=0;p<=tal;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|mi[j-2];
             if ((p&a[j])==a[j]) //a[j]记录的是j这个点之前必须要访问那些点，如果这些点都访问过了，当前状态才可以递推
              if (f[nxt][j]==-1||f[nxt][j]>f[p][i]+dis[i][j])
               f[nxt][j]=f[p][i]+dis[i][j];//f[nxt][j]表示的是到达J这个点状态为nxt的最短路
          }
       }
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for (int i=1;i<=m;i++)
     {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
     }
    for (int i=1;i<=k+1;i++)
     spfa(i);
    mi[0]=1;
    for (int i=1;i<30;i++)
     mi[i]=mi[i-1]<<1;
    scanf("%d",&t);
    for (int i=1;i<=t;i++)
     {
        int x,y;
        scanf("%d%d",&x,&y);
        a[y]+=mi[x-2];
     }
    memset(f,-1,sizeof(f));
    f[0][1]=0;
    tal=mi[k]-1;  
    dp();
    int ans=inf;
    for (int i=1;i<=k+1;i++)
     if (f[tal][i]!=-1)
      ans=min(ans,f[tal][i]+len[i]);
    printf("%d\n",ans);
}