POJ 3169 Layout

原题链接

思路：
转化为有负边的求最短路径问题，用Bellman-Ford算法
将每头牛当作一个顶点，互相亲近的牛由小编号指向大编号的边权值为正，互相厌恶的牛由大编号指向小编号的边权值为负
由i+1号牛指向i号牛的边权值为0
求此图中的最短路径
PS：从起点到j点沿箭头方向的路径上的权值和是第一头牛与j牛的最大距离，因此在所有由起点到j点的路径中，必须选取最小的一个，才能保证所有的路径都不超过其最大距离

AC代码：

#include <iostream>
#include <cstdio> 
#include <algorithm>
#include <cstdlib>
#include <cstring>
#define INF 10000000
using namespace std;

//《挑战》 P110

struct edge{
    int from,to,cost;
}e[100000];
int N,ML,MD;  //总牛数，互相亲近的牛的组数，互相厌恶的牛的组数
int d[1002];

int main()
{
    int i,j;
    scanf("%d %d %d",&N,&ML,&MD);
    getchar();
    for(i=0;i<ML;i++){
        scanf("%d %d %d",&(e[i].from),&(e[i].to),&(e[i].cost));
        getchar();
        e[i].from-=1;
        e[i].to-=1;
    }
    for(;i<MD+ML;i++){
        scanf("%d %d %d",&(e[i].to),&(e[i].from),&(e[i].cost));
        getchar();
        e[i].from-=1;
        e[i].to-=1;
        e[i].cost=-e[i].cost;
    }
    for(j=1;j<N;j++,i++){
        e[i].from=j;
        e[i].to=j-1;
        e[i].cost=0;
    }
    bool negative_loop=false;
    fill(d,d+N,INF);
    d[0]=0;
    for(i=0;i<N;i++){
        for(j=0;j<MD+ML+N-1;j++){
            int u=e[j].from;
            int v=e[j].to;
            if(d[u]<INF && d[v]>d[u]+e[j].cost){
                d[v]=d[u]+e[j].cost;
                if(i==N-1) negative_loop=true;
            }
        }
    }
    if(negative_loop)  d[N-1]=-1;
    else if(d[N-1]>=INF)  d[N-1]=-2;
    printf("%d\n",d[N-1]);
    return 0;
}