带负权的单源最短路(Bellman-Ford&&SPFA)

带负权的单源最短路(Bellman-Ford&&SPFA)

这次终于遇到带负权的单源最短路了,以前遇到的都是权值为正的情况,直接跑个Dijkstra算法即可

这次也终于能好好学学Bellman-Ford算法和SPFA算法了(Bellman-Ford算法的优化(用队列))

下面的博客,感觉很好,代码简单,而且用结构体和vector都有模板

https://blog.csdn.net/AAMahone/article/details/90085637

给道例题:
https://nanti.jisuanke.com/t/41305

就是今天的网络赛题签到题,弱鸡只会做签到题(呜呜呜)(The Preliminary Contest for ICPC Asia Nanjing 2019

开始队友准备开其他题,后面我看榜发现这题似乎是个签到题,于是就开始看题,由于太菜,看了很久才知道原来就是一个最短路的问题,比赛结束之后自己又好好做了这题,用了四种方式都尝试了下:

Bellman-Ford算法实现:(vector表示)

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
using namespace std;
const int maxn=550;
const int inf=0xffffff0;
typedef pair<int,int> pii;
int dis[maxn];
vector<pii> e[maxn];
void init(int  n)
{
    for(int i=0;i<n;i++)
        e[i].clear();
}
bool Bellman_Ford(int n,int st)
{
    for(int i=0;i<=n;i++)
    {
        dis[i]=inf;
    }
    dis[st]=0;
    int len;
    for(int k=0;k<n-1;k++)
    {
        for(int i=0;i<=n;i++)
        {
            len=e[i].size();
            for(int j=0;j<len;j++)
            {
                if(dis[e[i][j].first]>dis[i]+e[i][j].second)
                {
                    dis[e[i][j].first]=dis[i]+e[i][j].second;
                }
            }
        }
    }
    for(int i=0;i<=n;i++)
    {
        len=e[i].size();
        for(int j=0;j<len;j++)
        {
            if(dis[e[i][j].first]>dis[i]+e[i][j].second)
                return false;
        }
    }
    return true;
}
int main()
{
    int t;
    int n,m;
    int u,v,w;
    int a,b;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init(n);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            e[u].push_back(make_pair(v,w));
        }
        for(int i=0;i<6;i++)
        {
            scanf("%d%d",&a,&b);
            if(Bellman_Ford(n,b))
            {
                dis[a]=0-dis[a];
                printf("%d\n",dis[a]);
                e[a].push_back(make_pair(b,dis[a]));
            }
        }
    }
    return 0;
}

在这里插入图片描述

Bellman-Ford算法实现:(结构体数组)

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn=550;
const int inf=0xffffff0;
struct node
{
    int v,w,nxt;
}e[maxn];
int dis[maxn];
int head[maxn];
int tot=0;
void init()
{
    tot=0;
    memset(head,-1,sizeof(head));
}
void add(int u,int v,int w)
{
    e[tot].v=v;
    e[tot].w=w;
    e[tot].nxt=head[u];
    head[u]=tot++;
}
bool Bellman_Ford(int n,int st)
{
    for(int i=0;i<=n;i++)
        dis[i]=inf;
    dis[st]=0;
    for(int k=0;k<n-1;k++)
    {
        for(int i=0;i<=n;i++)
        {
            for(int j=head[i];j!=-1;j=e[j].nxt)
            {
                if(dis[e[j].v]>dis[i]+e[j].w)
                    dis[e[j].v]=dis[i]+e[j].w;
            }
        }
    }
    for(int i=0;i<=n;i++)
    {
        for(int j=head[i];j!=-1;j=e[j].nxt)
        {
            if(dis[e[j].v]>dis[i]+e[j].w)
            {
                return false;
            }
        }
    }
    return true;
}
int main()
{
    int t;
    int n,m;
    int a,b;
    int u,v,w;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
        }
        for(int i=0;i<6;i++)
        {
            scanf("%d%d",&a,&b);
            if(Bellman_Ford(n,b))
            {
                dis[a]=0-dis[a];
                printf("%d\n",dis[a]);
                add(a,b,dis[a]);
            }
        }
    }
    return 0;
}

在这里插入图片描述

SPFA算法实现:(vector)

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=550;
const int inf=0xffffff0;
typedef pair<int,int>pii;
vector<pii>e[maxn];
int dis[maxn];
int head[maxn];
int cnt[maxn];
int vis[maxn];
void init(int n)
{
    for(int i=0;i<=n;i++)
        e[i].clear();
}
bool SPFA(int n,int st)
{
    for(int i=0;i<=n;i++)
    {
        dis[i]=inf;
        vis[i]=0;
        cnt[i]=0;
    }
    dis[st]=0;
    int cur,len;
    queue<int>q;
    q.push(st);
    vis[st]=1;
    cnt[st]++;
    while(!q.empty())
    {
        cur=q.front();
        q.pop();
        vis[cur]=0;
        len=e[cur].size();
        for(int i=0;i<len;i++)
        {
            if(dis[e[cur][i].first]>dis[cur]+e[cur][i].second)
            {
                if(!vis[e[cur][i].first])
                {
                    cnt[e[cur][i].first]++;
                    q.push(e[cur][i].first);
                    vis[e[cur][i].first]=1;
                    if(cnt[e[cur][i].first]>=n)
                        return false;
                }
                dis[e[cur][i].first]=dis[cur]+e[cur][i].second;
            }
        }
    }
    return true ;
}
int  main()
{
    int t;
    int n,m;
    int a,b;
    int u,v,w;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init(n);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            e[u].push_back(make_pair(v,w));
        }
        for(int i=0;i<6;i++)
        {
            scanf("%d%d",&a,&b);
            if(SPFA(n,b))
            {
                dis[a]=0-dis[a];
                printf("%d\n",dis[a]);
                e[a].push_back(make_pair(b,dis[a]));
            }
        }
    }
    return 0;
}

在这里插入图片描述
SPFA算法:(结构体数组)

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=550;
const int inf=0xffffff0;
struct node
{
    int v,w,nxt;
}e[maxn];
int dis[maxn],vis[maxn],cnt[maxn],head[maxn],tot=0;
void init()
{
    tot=0;
    memset(head,-1,sizeof(head));
}void add(int u,int v,int w)
{
    e[tot].v=v;
    e[tot].w=w;
    e[tot].nxt=head[u];
    head[u]=tot++;
}
bool SPFA(int n,int st)
{
    for(int i=0;i<=n;i++)
    {
        dis[i]=inf;
        vis[i]=0;
        cnt[i]=0;
    }
    dis[st]=0;
    int cur;
    queue<int>q;
    q.push(st);
    vis[st]=1;
    cnt[st]++;
    while(!q.empty())
    {
        cur=q.front();
        q.pop();
        vis[cur]=0;
        for(int i=head[cur];i!=-1;i=e[i].nxt)
        {
            if(dis[e[i].v]>dis[cur]+e[i].w)
            {
                if(!vis[e[i].v])
                {
                    cnt[e[i].v]++;
                    q.push(e[i].v);
                    vis[e[i].v]=1;
                    if(cnt[e[i].v]>=n)
                    {
                        return false;
                    }
                }
                dis[e[i].v]=dis[cur]+e[i].w;
            }
        }
    }
    return true;
}
int main()
{
    int t;
    int n,m;
    int a,b;
    int u,v,w;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
        }
        for(int i=0;i<6;i++)
        {
            scanf("%d%d",&a,&b);
            if(SPFA(n,b))
            {
                dis[a]=0-dis[a];
                printf("%d\n",dis[a]);
                add(a,b,dis[a]);
            }
        }
    }
    return 0;
}

在这里插入图片描述

发布了127 篇原创文章 · 获赞 32 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/boliu147258/article/details/100188878