Wormholes（判斷是否存在負環）

鍊式前向星+spfa判斷負權

該題題解：

判斷有沒有負權産生，即從一個點到一個點的權值是否為負

題目：

該題題意：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define maxn 10005
using namespace std;
struct node
{
    int to;//終點
    int next;//尋址
    int val;//權值
}graph[maxn];
int head[maxn];
int book[maxn];
int dis[maxn];
int n,m,w,cnt;//n個點,m條雙向邊,w條單向蟲洞

void inputs(int u,int v,int w)
{     //scanf("%d %d",&u,&v);
        graph[cnt].to=v;
        graph[cnt].val=w;
        graph[cnt].next=head[u];
        head[u]=cnt;
        cnt++;
}

bool spfa(int s)
{
    queue<int>q;
    memset(book,0,sizeof(book));
    memset(dis,0x3f3f3f3f,sizeof(dis));
    dis[s]=0;
    //book[s]=0;
    q.push(s);//入隊
    while(!q.empty())
    {
        int st=q.front();
        q.pop();
        book[st]=0;
        for(int k=head[st];k!=-1;k=graph[k].next)
        {
            int to=graph[k].to;
            int w=graph[k].val;
            if(dis[to]>dis[st]+w)
            {

                dis[to]=dis[st]+w;
                if(book[to]==0)
                {
                    book[to]==1;
                    q.push(to);

                }
            }
        }
        if(dis[1]<0)//在入隊的過程中回到了1的位置的點是否有小于0的
            return true;
    }
    return false;
}
int main()
{

    int t;
    scanf("%d",&t);
    while(t--)
    {
        cnt=0;
        memset(head,-1,sizeof(head));
        scanf("%d %d %d",&n,&m,&w);
        while(m--)
        {
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            inputs(u,v,w);
            inputs(v,u,w);
        }
        while(w--)
        {   int u,v,w;
             scanf("%d %d %d",&u,&v,&w);
             inputs(u,v,-w);
        }
        if(spfa(1))
        {
            printf("YES\n");

        }
        else
            printf("NO\n");
    }
}