DAG的最短路

最新推荐文章于 2021-12-07 22:15:45 发布

原创最新推荐文章于 2021-12-07 22:15:45 发布 · 574 阅读

1 ·

CC 4.0 BY-SA版权

数据结构同时被 2 个专栏收录

23 篇文章

订阅专栏

动态规划

18 篇文章

订阅专栏

本文通过一个具体的竞赛题目，介绍了如何在有向无环图(DAG)上使用SPFA算法解决最短路径问题。特别关注了当存在负权边时，Dijkstra算法的不适用性，以及如何利用DAG的拓扑排序特性来优化SPFA算法，使其在规定时间内找到最优解。

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

题目：https://ac.nowcoder.com/acm/contest/329/B

分析：由于带负权，肯定不能用Dijkstra,用spfa可高效解决，但是特殊 DAG 的性质使得 SPFA 算法无法在规定的时间限内求解出答案，此时由DAG的拓扑序列可以很好解决该问题。

因此，放弃vis数组，以拓扑序列来入队

Ac code:

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
typedef long long ll;
const ll INF=1e10;
vector<pair<int,ll> >edge1[maxn];
vector<pair<int,ll> >edge2[maxn];
ll dis1[maxn],dis2[maxn];
int dg1[maxn],dg2[maxn];
int n;
ll Spfa(int s,ll dis[],int dg[],vector<pair<int,ll> >edge[])
{
    queue<int>q;
    while(!q.empty()) q.pop();
    for(int i=1;i<=n;i++) dis[i]=INF;
    dis[s]=0;
    for(int i=1;i<=n;i++) if(dg[i]==0) q.push(i);
    while(!q.empty()){
        int u=q.front();
        q.pop();
        for(int i=0;i<edge[u].size();i++){
            int v=edge[u][i].first;
            ll w=edge[u][i].second;
            dis[v]=min(dis[v],dis[u]+w);
            if(--dg[v]==0)
                q.push(v);
        }
    }
    return max(0ll,dis[n]);
}
int main()
{
    int t,m;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=0;i<=n;i++) edge1[i].clear(),edge2[i].clear();
        int u,v;
        ll w0,w1,w2;
        for(int i=1;i<=m;i++){
           scanf("%d%d%lld%lld%lld",&u,&v,&w0,&w1,&w2);
           edge1[u].push_back(make_pair(v,-w1+w0)),edge2[u].push_back(make_pair(v,-w2+w0));
           dg1[v]++,dg2[v]++;
        }
        ll ans1=Spfa(1,dis1,dg1,edge1);
        ll ans2=Spfa(1,dis2,dg2,edge2);
        if(ans1<ans2){
            puts("cnznb!!!");
            printf("%lld\n",ans2-ans1);
        }
        else if(ans1>ans2){
            puts("rip!!!");
            printf("%lld\n",ans1-ans2);
        }
        else puts("oof!!!");
    }
    return 0;
}