【2019牛客暑期多校训练营 第四场 J题】【free】【最短路+DP】

本文探讨了在无向图中寻找从起点到终点消除K条边的最短路径问题,提供了两种解法:一是使用动态规划,二是先求最短路径再调整。代码实现展示了优先队列和邻接表的应用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目链接:
https://ac.nowcoder.com/acm/contest/884/J
题意:
给一个无向图,求从起点S到达终点T最多可以消除K条边的最小路径长度。
题解:
解①:
这显然是一个DP题,设f[i][j]f[i][j]f[i][j]为从S到达iii点消除jjj条边的最小路径长度,
状态转移方程为:f[to][j]=min(f[to][j],min(f[from][j]+dis[from][to],f[from][j−1]))f[to][j] = min(f[to][j], min(f[from][j] + dis[from][to], f[from][j-1]))f[to][j]=min(f[to][j],min(f[from][j]+dis[from][to],f[from][j1]))
边跑最短路边DP即可

解②:
这里还有种做法是直接跑最短路,然后将去掉最短路径上的K条路,剩下的路径求和即为答案。这个一开始想过,还以为是错的就想用DP来做,没有证明过。
代码:
解法1:

const int MAX = 1e3 + 10;
 
int N, M, S, T, K;
int dis[MAX][MAX];
 
struct edge {
    int nxt, to, w;
}e[MAX * 2];
 
int head[MAX], tot;
 
void add(int u, int v, int w) {
    e[++tot].nxt = head[u];
    e[tot].to = v;
    e[tot].w = w;
    head[u] = tot;
}
 
struct node {
    int u, d, k;
    bool operator <(const node& rhs) const {
        if (d == rhs.d)return k > rhs.k;
        return d > rhs.d;
    }
};
priority_queue<node> q;
 
 
void dijkstra(int s) {
    memset(dis, 0x3f, sz(dis));
    dis[s][0] = 0;
    q.push(node{ s,0,0 });
    while (!q.empty()) {
        node u = q.top(); q.pop();
        int k = u.k, now = u.u;
        for (int i = head[now]; i; i = e[i].nxt) {
            int to = e[i].to, w = e[i].w;
            if (dis[now][k] + w < dis[to][k])
                dis[to][k] = dis[now][k] + w, q.push(node{ to, dis[to][k], k });
            if (k + 1 <= K && dis[now][k] < dis[to][k + 1])
                dis[to][k + 1] = dis[now][k], q.push(node{ to, dis[to][k + 1], k + 1 });
        }
    }
}
 
int main() {
    tot = 0;
    scanf("%d%d%d%d%d", &N, &M, &S, &T, &K);
    for (int i = 1; i <= M; i++) {
        int u, v, w;
        scanf("%d%d%d", &u, &v, &w);
        if (u == v)continue;
        add(u, v, w);
        add(v, u, w);
    }
    dijkstra(S);
    int ans = INF;
    for (int i = 0; i <= K; i++)ans = min(ans, dis[T][i]);
    printf("%d\n", ans);
    return 0;
}

解法2:

const int MAX = 1e3 + 10;
 
int N, M, S, T, K;
int dis[MAX], vis[MAX], pre[MAX], cost[MAX];
 
struct edge {
    int nxt, to, w;
}e[MAX * 2];
 
int head[MAX], tot;
 
void add(int u, int v, int w) {
    e[++tot].nxt = head[u];
    e[tot].to = v;
    e[tot].w = w;
    head[u] = tot;
}
 
struct node {
    int u, d;
    bool operator <(const node& rhs) const {
        return d > rhs.d;
    }
};
priority_queue<node> q;
 
 
void dijkstra(int s) {
    memset(dis, 0x3f, sz(dis));
    dis[s] = 0;
    q.push(node{ s,0 });
    while (!q.empty()) {
        node u = q.top(); q.pop();
        int now = u.u;
        if (vis[now])continue;
        vis[now] = 1;
        for (int i = head[now]; i; i = e[i].nxt)
            if (dis[now] + e[i].w < dis[e[i].to]) {
                dis[e[i].to] = dis[now] + e[i].w;
                pre[e[i].to] = now;
                cost[e[i].to] = e[i].w;
                if (!vis[e[i].to])
                    q.push(node{ e[i].to, dis[e[i].to] });
            }
    }
}
 
priority_queue<int, vector<int>, greater<int> > qq;
 
int main() {
    tot = 0;
    scanf("%d%d%d%d%d", &N, &M, &S, &T, &K);
    for (int i = 1; i <= M; i++) {
        int u, v, w;
        scanf("%d%d%d", &u, &v, &w);
        if (u == v)continue;
        add(u, v, w);
        add(v, u, w);
    }
    dijkstra(S);
    for (int i = T; i != S; i = pre[i])qq.push(cost[i]);
    int cnt = max(0, (int)qq.size() - K), ans = 0;
    while (cnt--) {
        ans += qq.top(); qq.pop();
    }
    printf("%d\n", ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值