POJ 3169 Layout

两个关系好的牛之间不能超过一定的距离，两个关系不好的牛之间的距离不能少一定的距离。如果存在负还输出-1，牛1和牛N不连通 输出-2，否则输出他们的之间距离。

#include<cstdio>
#include<cstring>
#include<cstdlib>

struct N
{
    int u,v,w;
} edge[20010];

int dis[1010];
int INF = (1<<29);

int bellman_ford(int n,int e)
{
    int flag;
    int i,j,k,u,v,w;
    for(i = 1;i < n; i++)
    {
        flag = 1;
        for(j = 0;j < e; j++)
        {
            u = edge[j].u;
            v = edge[j].v;
            w = edge[j].w;
            if(dis[u] + w < dis[v])
            {
                dis[v] = dis[u] + w;
                flag = 0;
            }
        }
        if(flag) break;
    }
    for(j = 0;j < e; j++)
    {
         u = edge[j].u;
            v = edge[j].v;
            w = edge[j].w;
            if( dis[u] + w < dis[v])
            {
                return 1;
            }
    }
    return 0;
}

int main()
{
    int n,m,l,i,e,u,v,w;
   scanf("%d %d %d",&n,&m,&l);
    
        for(i = 1; i <= n; i++)
        {
            dis[i] = INF;
        }
        dis[1] = 0;
        for(e = 0, i = 0; i < m; i++)
        {
            scanf("%d %d %d",&u,&v,&w);
            edge[e].u = u;
            edge[e].v = v;
            edge[e].w = w;
            e++;
        }
        for(i = 0; i < l; i++)
        {
            scanf("%d %d %d",&u,&v,&w);
            edge[e].u = v;
            edge[e].v = u;
            edge[e].w = -w;
            e++;
        }

   

        if(bellman_ford(n,e))
            printf("-1\n");
        else
        {
            if(dis[n] == INF)
                printf("-2\n");
            else printf("%d\n",dis[n]);
        }
    
    
    return 0;
}