POJ3259 Wormholes(Bellmanford判断负环）

刚刚学了Bellmanford算法，于是找了道做过的模板题来试试手。当时直接用的模板，今天靠理解手打了一遍Bellmanford，进一步加深了理解。
Bellmanford算法，简而言之就是对一张有n个点的图的所有边，进行n-1次松弛操作。然后判断，如果仍然能进行第n次松弛操作，说明图里有负环，可以沿着这个负环不停地走下去让最短路径变为无穷小。
建图用的链式前向星。
前M条权值为正的边是双向边，后W条权值为负的是单向的。

#include <iostream>#include <cstring>#include <cstdio>#include <cmath>#include <vector>#include <algorithm>using namespace std;const int INF=0x3f3f3f3f;const int MAXN=1000;int dis[MAXN];int head[MAXN];struct edge{    int to,len,next;}E[8000];bool bellmanford(int n){    for(int i=2;i<=n;i++)   dis[i]=INF;    dis[1]=0;    for(int i=1;i<=n-1;i++)                 //对链式前向星中共2M+W条边各执行n-1次松弛操作        {            for(int j=1;j<=n;j++){                if(dis[j]==INF)     continue;                for(int k=head[j];k!=0;k=E[k].next){                        //链式前向星编号从1开始，故以K！=0结束，若从0开始，改为k！=-1结束                    if(E[k].len!=INF&&dis[E[k].to]>(dis[j]+E[k].len)){                        dis[E[k].to]=dis[j]+E[k].len;                    }                }            }        }        for(int j=1;j<=n;j++){                                            //若经过以上n-1次松弛以后仍然能松弛，说明图中存在权值为负的环，无最短路径            for(int k=head[j];k!=0;k=E[k].next){                if(E[k].len!=INF&&dis[E[k].to]>(dis[j]+E[k].len)){                   return false;                                           //存在负环                }            }        }        return true;                                                    //无负环}int main(){    int F,N,M,W,i,j,k;    scanf("%d",&F);    while(F--)    {   memset(head,0,sizeof(head));        scanf("%d%d%d",&N,&M,&W);        memset(E,0,sizeof(E));        memset(dis,0,sizeof(dis));        for(i=1;i<=M;i++)                                    //建立了编号从1到2M，共2M条边（前M条与后M条一一对应为反向边）        {   int index;            scanf("%d%d%d",&index,&E[i].to,&E[i].len);            E[i].next=head[index];            head[index]=i;            E[i+M].to=index;            E[i+M].len=E[i].len;            E[i+M].next=head[E[i].to];            head[E[i].to]=i+M;        }        for(i=2*M+1;i<=2*M+W;i++)                                           {   int index;            scanf("%d%d%d",&index,&E[i].to,&E[i].len);    //建立了编号从2M+1到2M+W共W条边            E[i].len=E[i].len*(-1);            E[i].next=head[index];            head[index]=i;        }      /* for(int j=1;j<=N;j++){                                            //测试，检查建边情况            for(int k=head[j];k!=0;k=E[k].next){                    cout<<j<<" "<<E[k].to<<" "<<E[k].len<<endl;                                                     }            }       */        if(bellmanford(N))        {            printf("NO\n");        }        else             printf("YES\n");    }    return 0;}