Test 3.27 T2 旅行

Posted 2021-11-25 lslzf

Description

FGD 想从成都去上海旅游。在旅途中他希望经过一些城市并在那里欣赏风景,品尝风味小吃或者做其他的有趣的事情。经过这些城市的顺序不是完全随意的,比如说 FGD 不希望在刚吃过一顿大餐之后立刻去下一个城市登山,而是希望去另外什么地方喝下午茶。幸运的是,FGD 的旅程不是既定的,他可以在某些旅行方案之间进行选择。由于 FGD 非常讨厌乘车的颠簸,他希望在满足他的要求的情况下,旅行的距离尽量短,这样他就有足够的精力来欣赏风景。整个城市交通网络包含 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 , 总长度为 19。注意 FGD 为了从城市 2 到城市 4 可以路过城市 3,但不在城市 3 停留。这样就不违反 FGD 的要求了。并且由于 FGD 想要走最短的路径,因此这个方案正是 FGD需要的。

Input

第一行包含 3 个整数 N(2<=N<=20000),M(1<=M<=200000),K(0<=K<=20),意义如上所述。以下 M 行,每行包含 3 个整数 X,Y,Z,(1<=X<y<=n, 1<="Z=1000)表示城市 X与 Y 之间有一条双向道路。你可以认为输入文件使得一定能自成都到上海以及任何 FGD 想要去的城市。"下一行包含一个整数 G(0<=G<=K*(K-1)/2)。以下 G 行,每行包含 2 个整数 X Y,(2<=X,Y<=K+1)表示 FGD 想要在游览城市Y 之前,一定要游览城市 X。你可以认为至少存在一种满足所有限制的游览方案。

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

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

Solution

看到k非常小，就可以想到状态压缩的动态规划(先例：售货员的难题)。设f[i][j]表示经过的1到k+1的点的集合为i，当前所在点为j时的最短路径。那么有如下状态转移方程：
\\[ f[i|(1<<l)][l]=min(f[i][j]+dis[j][l],f[i|(1<<<l)][l]) \\]
其中dis[j][l]表示从j到l的最短路径，那么为了判断一个点是否满足其先要访问的点的条件，可以用g[i]表示一个点i的所有前驱节点集合(同样用状态压缩)，如果在状态转移方程中满足i&g[i]=g[i]，就说明可行 。dis数组可以用SPFA跑出来。动态规划完成后，由于没有将第n个点计算在内，所以最终的答案为：
\\[ ans=max(f[1<<<k][i]+dis[i][n])_{i=1,2,...,k+1} \\]
另外，为了表示方便和节省空间，第i个点在二进制集合中的位置可设为i-2，这样就可以从第0位开始表示了。

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
#define N 20002
#define M 200002
#define int long long
using namespace std;
int head[N],ver[M*2],nxt[M*2],edge[M*2],l;
int n,m,k,c,i,j,d[22][22],dis[N],f[1048576][22],g[22];
bool in[N];
int read()
{
    char c=getchar();
    int w=0;
    while(c<'0'||c>'9') c=getchar();
    while(c<='9'&&c>='0'){
        w=w*10+c-'0';
        c=getchar();
    }
    return w;
}
void insert(int x,int y,int z)
{
    l++;
    ver[l]=y;
    edge[l]=z;
    nxt[l]=head[x];
    head[x]=l;
}
void SPFA(int s)
{
    queue<int> q;
    memset(in,0,sizeof(in));
    memset(dis,0x3f,sizeof(dis));
    q.push(s);
    in[s]=1;
    dis[s]=0;
    while(!q.empty()){
        int x=q.front();
        q.pop();
        for(int i=head[x];i;i=nxt[i]){
            int y=ver[i];
            if(dis[y]>dis[x]+edge[i]){
                dis[y]=dis[x]+edge[i];
                if(!in[y]){
                    q.push(y);
                    in[y]=1;
                }
            }
        }
        in[x]=0;
    }
    for(int i=1;i<=k+1;i++) d[s][i]=dis[i];
    d[s][k+2]=dis[n];
}
signed main()
{
    freopen("atr.in","r",stdin);
    freopen("atr.out","w",stdout);
    n=read();m=read();k=read();
    for(i=1;i<=m;i++){
        int u,v,w;
        u=read();v=read();w=read();
        insert(u,v,w);
        insert(v,u,w);
    }
    c=read();
    for(i=1;i<=c;i++){
        int x,y;
        x=read();y=read();
        g[y]|=(1<<(x-2));
    }
    for(i=1;i<=k+1;i++) SPFA(i);
    memset(f,-1,sizeof(f));
    f[0][1]=0;
    for(i=0;i<=(1<<k)-1;i++){
        for(j=1;j<=k+1;j++){
            if(f[i][j]!=-1){
                for(l=2;l<=k+1;l++){
                    if((i&g[l])==g[l]){
                        if(f[i|(1<<(l-2))][l]==-1) f[i|(1<<(l-2))][l]=f[i][j]+d[j][l];
                        else f[i|(1<<(l-2))][l]=min(f[i|(1<<(l-2))][l],f[i][j]+d[j][l]);
                    }
                }
            }
        }
    }
    int ans=1<<30;
    for(i=1;i<=k+1;i++){
        if(f[(1<<k)-1][i]!=-1) ans=min(ans,f[(1<<k)-1][i]+d[i][k+2]);
    }
    cout<<ans<<endl;
    fclose(stdin);
    fclose(stdout);
    return 0;
}