Vijos 1053 Easy sssp

Vijos 1053 Easy sssp

地址 https://vijos.org/p/1053

描述

输入数据给出一个有N(2 <= N <= 1,000)个节点，M(M <= 100,000)条边的带权有向图.
要求你写一个程序, 判断这个有向图中是否存在负权回路. 如果从一个点沿着某条路径出发, 又回到了自己, 而且所经过的边上的权和小于0, 就说这条路是一个负权回路.
如果存在负权回路, 只输出一行-1;
如果不存在负权回路, 再求出一个点S(1 <= S <= N)到每个点的最短路的长度. 约定: S到S的距离为0, 如果S与这个点不连通, 则输出NoPath.

格式

输入格式

第一行: 点数N(2 <= N <= 1,000), 边数M(M <= 100,000), 源点S(1 <= S <= N);
以下M行, 每行三个整数a, b, c表示点a, b(1 <= a, b <= N)之间连有一条边, 权值为c(-1,000,000 <= c <= 1,000,000)

输出格式

如果存在负权环, 只输出一行-1, 否则按以下格式输出
共N行, 第i行描述S点到点i的最短路:
如果S与i不连通, 输出NoPath;
如果i = S, 输出0;
其他情况输出S到i的最短路的长度.

样例输入

6 8 1
1 3 4
1 2 6
3 4 -7
6 4 2
2 4 5
3 6 3
4 5 1
3 5 4

样例输出
样例1
样例输入1

6 8 1
1 3 4
1 2 6
3 4 -7
6 4 2
2 4 5
3 6 3
4 5 1
3 5 4
样例输出1[复制]

0
6
4
-3
-2
7

限制

Test5 5秒
其余 1秒

图片帮助

之题乍一看起来十分水，有负权，用SPFA
但仔细思考我们发现了一件事情，没说只有一个连通块
如果出现下图这样的话直接用SPFA会崩掉的！

该输出“NoPath.”的点让我们输出了一个无限大的的距离，怎么办？

其实看起来也不难，我们只需将每一个点都进队扫一遍，这样对连通块进行一次判断

#include <stdio.h>
#include <string.h>
#include <queue>
#include <iostream>
using namespace std;
int n,m,s,ans;
int tot,head[100010],to[100010],val[100010],nex[100010];
int f[100010],v[100010],ns[100010],c[1000010];
void add(int x,int y,int z)
{
    to[++tot]=y;
    val[tot]=z;
    nex[tot]=head[x];
    head[x]=tot;
}
int spfa(int z )
{
    int hehe=0;
    memset(f,0x3f,sizeof(f));
    memset(v,0,sizeof(v));
    memset(ns,0,sizeof(ns));
    queue<int> q;
    q.push(z),f[z]=0,v[z]=1;
    while(q.size())
    {
        int x,y;
        x=q.front(),q.pop(),v[x]=0;
        for(int i=head[x];i;i=nex[i])
            if(f[y=to[i]]>f[x]+val[i])
            {
                f[y]=f[x]+val[i];
                if(v[y]==0)
                {
                     q.push(y);
                     v[y]=1;
                     ns[y]++;
                     if(ns[y]>n)
                     {
                        hehe=1;
                        return hehe;
                     }
                }
            }
    }
    if(z==s)
        for(int i=0;i<=n;i++)
            c[i]=f[i];
    return hehe;
}
int main()
{
    scanf("%d%d%d",&n,&m,&s);
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    ans=spfa(s);
    if(ans==1)
    {
        printf("-1\n");
        return 0;
    }
    for(int i=1;i<=n;i++)
        if(ns[i]==0);
            ans=spfa(i);
    if(ans==1)
    {
        printf("-1\n");
        return 0;
    }
    for(int i=1;i<=n;i++)
    {
        if(f[i]>1000000000)
            printf("NoPath\n");
        else 
            printf("%d\n",c[i]);
    }
    return 0;
}

但是

WA的还WA，新增了TLE
何解？

于是我们做一下优化，只将从来没有进过队列的点在扫一遍，同时使用两个SPFA，分别判断距离与是否连通

C++代码如下

#include <stdio.h>
#include <string.h>
#include <queue>
#include <iostream>
using namespace std;
int n,m,s,ans;
int tot,head[100010],to[100010],val[100010],nex[100010];
long long f[100010];
int v[100010],se[100010];
void add(int x,int y,int z)
{
    to[++tot]=y;
    val[tot]=z;
    nex[tot]=head[x];
    head[x]=tot;
}
int checkspfa(int s)//检查是否有负权环
{
    memset(f,0x3f,sizeof(f));
    memset(v,0,sizeof(v));
    queue<int>q;
    q.push(s),v[s]=1,f[s]=0;
    while(q.size())
    {
        int x;
        x=q.front(),q.pop(),v[x]=0;
        for(int i=head[x];i;i=nex[i])
        {
            if(f[to[i]]>f[x]+val[i])
            {
                f[to[i]]=f[x]+val[i];
                se[to[i]]++;
                q.push(to[i]);
                if(se[to[i]]>n)
                    return 1;
            }
        }
    }
    return 0;
}
int spfa(int z )
{
    memset(f,0x3f,sizeof(f));
    memset(v,0,sizeof(v));
    queue<int> q;
    q.push(z),f[z]=0,v[z]=1;
    while(q.size())
    {
        int x,y;
        x=q.front(),q.pop(),v[x]=0;
        for(int i=head[x];i;i=nex[i])
            if(f[y=to[i]]>f[x]+val[i])
            {
                f[y]=f[x]+val[i];
                if(v[y]==0)
                {
                     q.push(y);
                     v[y]=1;
                }
            }
    }
    return 0;
}
int main()
{
    memset(se,0,sizeof(se));
    scanf("%d%d%d",&n,&m,&s);
    for(int i=1;i<=m;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    for(int i=1;i<=n;i++)
    {
        if(se[i]==0)//未被判断过
        {
            int ans=checkspfa(i);
            if(ans==1)
            {
                printf("-1");//存在负权环
                return 0;
            }
        }   
    }
    int jdshg=spfa(s);
    for(int i=1;i<=n;i++)
    {
        if(f[i]==0x3f3f3f3f3f3f3f3fll)//二者距离为最大值，既不连通
            printf("NoPath\n");
        else
            printf("%lld\n",f[i]);
    }
    return 0;
}

这样就AC了

PS：如果遇见让你不爽的 OIer 推荐给他这道题吧

题目传送门

https://vijos.org/p/1053