[luoguAT_abc369_e]Sightseeing Tour

题意

给定一个包含 N 个点和 M 条无向边的带权图,保证图中没有自环,但可能包含重边。

给出 Q 次查询,每次查询给出 K 条边 B1,B2,,BK,要求求出从节点 1 到节点 N 且这 K 条边都至少经过一次的最短路(经过边的方向和顺序任意)。

赛时 Dijkstra 状态压缩 [TLE]

赛后

由于 K 很小,因此我们可以枚举经过 K 条边的顺序和方向,提前预处理出 dist 数组即可直接计算出最短距离。

代码

#include <iostream>
#include <algorithm>
#include <cstring>
#define x first 
#define y second 

using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<int, PII> PIP;

const int N = 405, M = 200005;

LL dist[N][N];
PIP edges[M];
int n, m, q, k;
int b[10], perm[10];

int main(){
    scanf("%d%d", &n, &m);
    memset(dist, 0x3f, sizeof dist);
    for (int i = 1; i <= n; i ++ ) dist[i][i] = 0;
    for (int i = 1; i <= m; i ++ ){
        int u, v, w;
        scanf("%d%d%d", &u, &v, &w);
        dist[u][v] = dist[v][u] = min(dist[u][v], 1ll * w);
        edges[i] = {w, {u, v}};
    }

    for (int k = 1; k <= n; k ++ )
        for (int i = 1; i <= n; i ++ )
            for (int j = 1; j <= n; j ++ )
                dist[i][j] = min(dist[i][j], dist[i][k] + dist[k][j]);
    
    scanf("%d", &q);
    while (q -- ){
        scanf("%d", &k);
        for (int i = 1; i <= k; i ++ ) scanf("%d", &b[i]), perm[i] = i;

        LL ans = 0x3f3f3f3f3f3f3f3f;
        do {
            for (int state = 0; state < (1 << k); state ++ ){
                LL res = 0, last = 1;
                for (int i = 1; i <= k; i ++ ) {
                    PIP edge = edges[b[perm[i]]];
                    int from = edge.y.x, to = edge.y.y;
                    if ((state >> i - 1) & 1) swap(from, to);
                    res += dist[last][from] + edge.x;
                    last = to;
                }
                res += dist[last][n];
                ans = min(ans, res);
            }
        } while (next_permutation(perm + 1, perm + k + 1));

        printf("%lld\n", ans);
    }

    return 0;
}
posted @   是一只小蒟蒻呀  阅读(13)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 使用C#创建一个MCP客户端
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示