poj 3169 Layout （差分约束）

3169 -- Layout

　　继续差分约束。

　　这题要判起点终点是否连通，并且要判负环，所以要用到spfa。

　　对于ML的边，要求两者之间距离要小于给定值，于是构建(a)->(b)=c的边。同理，对于MD的，构建(b)->(a)=-c的边。然后就是(i+1)->(i)=0，两者距离大于0的限制。

代码如下：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>

using namespace std;

const int N = 1111;
const int M = 33333;
struct Edge {
    int t, nx, c;
    Edge() {}
    Edge(int t, int nx, int c) : t(t), nx(nx), c(c) {}
} edge[M];
int eh[N], ec;

void init() {
    memset(eh, -1, sizeof(eh));
    ec = 0;
}

void addedge(int s, int t, int c) {
    edge[ec] = Edge(t, eh[s], c);
    eh[s] = ec++;
}

int cnt[N], dis[N], n;
bool inq[N];
queue<int> q;
const int INF = 0x5fffffff;

int spfa(int s, int t) {
    while (!q.empty()) q.pop();
    for (int i = 1; i <= n; i++) inq[i] = false, dis[i] = INF, cnt[i] = n + 1;
    q.push(s);
    dis[s] = 0;
    inq[s] = true;
    cnt[s]--;
    while (!q.empty()) {
        int cur = q.front();
        // cout << cur << ' ' << eh[cur] << endl;
        q.pop();
        inq[cur] = false;
        for (int i = eh[cur]; ~i; i = edge[i].nx) {
            Edge &e = edge[i];
            //cout << dis[e.t] << ' ' << dis[cur] + e.c << endl;
            if (dis[e.t] > dis[cur] + e.c) {
                dis[e.t] = dis[cur] + e.c;
                if (cnt[e.t]) cnt[e.t]--;
                else return -1;
                if (!inq[e.t]) {
                    q.push(e.t);
                    inq[e.t] = true;
                }
            }
        }
    }
    if (dis[t] == INF) return -2;
    else return dis[t];
}

int main() {
    int ml, md;
    int x, y, c;
    while (~scanf("%d%d%d", &n, &ml, &md)) {
        init();
        for (int i = 0; i < ml; i++) {
            scanf("%d%d%d", &x, &y, &c);
            addedge(x, y, c);
        }
        for (int i = 0; i < md; i++) {
            scanf("%d%d%d", &x, &y, &c);
            addedge(y, x, -c);
        }
        for (int i = 1; i < n; i++) {
            //cout << i + 1 << ' ' << i << endl;
            addedge(i + 1, i, 0);
        }
        printf("%d
", spfa(1, n));
    }
    return 0;
}

——written by Lyon