牛客CSP-S提高组赛前集训营5 - B 十二桥问题

最新推荐文章于 2022-09-25 00:03:27 发布

Nightmare丶

最新推荐文章于 2022-09-25 00:03:27 发布

阅读量244

点赞数

分类专栏：最短路状压dp 文章标签： dijkstra 状压dp

本文链接：https://blog.csdn.net/qq_43544481/article/details/102969534

版权

状压dp 同时被 2 个专栏收录

13 篇文章 0 订阅

订阅专栏

最短路

8 篇文章 0 订阅

订阅专栏

题目描述

小多所在的城市可以看成是有 $n$ 个点 $m$ 条边的无向图（结点从 $1$ 标号），每条边有一个距离 $d_i$ ，其中有 $k$ 条边是小希特别想走过的 $k$ 座大桥。

小多和小希现在呆在 $1$ 号结点，请你帮小多规划一条最短路线，使得小多和小希能从当前位置出发，并经过这 $k$ 座桥，最后回到结点 $1$ 。

题解：

这道题刚开始看就觉得和BZOJ 1097: [POI2007]旅游景点atr是类似的题，只是过程中经过的点，而这道题变成了经过的边，贴一篇题解。

还是一样的思路，我们先预处理出所有桥包含的端点，全部跑一遍最短路。

由于k很小，很容易想到状压，然后想压什么？压边还是压点，压边我们只有最多12条边，只需要2^12次方的状态，而压点我们最多可能有24个点，状态太多，所以我们只能选择压边。

设f[u][S] 表示当前在u点，已经经过的桥的集合为S。

然后转移只需要考虑最后在桥的哪一个点即可，即设桥有两点为u和v，我们经过此桥后当前可能是u点，也可能是v点，如果是u点，就说明我们是先跑到的v点，然后经过桥走到的u点，所以加上之前点道v点的最短距离即可。反之也是一样的理解。

AC代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int MAXN = 1e6+50;
const int INF = 0x3f3f3f3f;
map<int,int> mp;
priority_queue<pair<int,int> > que;
struct node{
    int v,w,nxt;
}edge[MAXN<<1];
int head[MAXN],tot,sz;
int id[MAXN],from[MAXN],to[MAXN],c[MAXN];
int dp[30][(1<<13)+50];
int dis[MAXN],vis[MAXN];
int g[30][30];
inline void add(int u,int v,int w){
    edge[++tot].v = v,edge[tot].w = w,edge[tot].nxt = head[u],head[u] = tot;
    edge[++tot].v = u,edge[tot].w = w,edge[tot].nxt = head[v],head[v] = tot;
}
inline void dijkstra(int now){
    int s = id[now];
    while(!que.empty()) que.pop();
    memset(dis,INF,sizeof(dis));
    memset(vis,0,sizeof(vis));
    dis[s] = 0;
    que.push(make_pair(0,s));
    while(!que.empty()){
        int u = que.top().second; que.pop();
        if(vis[u]) continue;
        vis[u] = 1;
        for(int i=head[u];i;i=edge[i].nxt){
            int v = edge[i].v,w = edge[i].w;
            if(dis[v] > dis[u] + w){
                dis[v] = dis[u] + w;
                que.push(make_pair(-dis[v],v));
            }
        }
    }
    for(int i=1;i<=sz;i++)
        g[now][i] = dis[id[i]];
}
signed main(){
    int n,m,k; scanf("%lld%lld%lld",&n,&m,&k);
    mp[1]=++sz; id[1]=1;
    for(int i=1;i<=m;i++){
        int u,v,w; scanf("%lld%lld%lld",&u,&v,&w);
        add(u,v,w);
        if(i<=k && !mp[u]) { mp[u]=++sz;id[sz]=u; }
        if(i<=k && !mp[v]) { mp[v]=++sz;id[sz]=v; }
        if(i<=k) from[i]=mp[u],to[i]=mp[v],c[i]=w;
    }
    for(int i=1;i<=sz;i++) dijkstra(i);
    memset(dp,INF,sizeof(dp));
    dp[1][0] = 0;
    for(int s=0;s<(1<<k);s++){
        for(int i=1;i<=sz;i++){
            if(dp[i][s]!=INF){
                for(int j=1;j<=k;j++){
                    if(!((s>>(j-1))&1)){
                        int u = from[j],v = to[j],w = c[j];
                        dp[u][s|(1<<j-1)] = min(dp[u][s|(1<<j-1)],dp[i][s]+g[i][v]+w);
                        dp[v][s|(1<<j-1)] = min(dp[v][s|(1<<j-1)],dp[i][s]+g[i][u]+w);
                    }
                }
                dp[1][s] = min(dp[1][s],dp[i][s]+g[i][1]);
            }
        }
    }
    printf("%lld\n",dp[1][(1<<k)-1]);
    return 0;
}