【洛谷】P1260 工程规划

题目地址：

https://www.luogu.com.cn/problem/P1260

题目描述：
造一幢大楼是一项艰巨的工程，它是由$n$个子任务构成的，给它们分别编号$1,2,\dots ,n(5\le n\le 1000)$。由于对一些任务的起始条件有着严格的限制，所以每个任务的起始时间$T_1,T_2,\dots ,T_n$并不是很容易确定的(但这些起始时间都是非负整数，因为它们必须在整个工程开始后启动)。例如：挖掘完成后，紧接着就要打地基；但是混凝土浇筑完成后，却要等待一段时间再去掉模板。这种要求就可以用$m(5\le m\le 5000)$个不等式表示，不等式形如$T_i-T_j\le b$代表$i$和$j$的起始时间必须满足的条件。每个不等式的右边都是一个常数$b$，这些常数可能不相同，但是它们都在区间$(-100，100)$内。你的任务就是写一个程序，给定像上面那样的不等式，找出一种可能的起始时间序列$T_1,T_2,\dots ,T_n$，或者判断问题无解。对于有解的情况，要使最早进行的那个任务和整个工程的起始时间相同，也就是说，$T_1,T_2,\dots ,T_n$中至少有一个为$0$。

输入格式：
第一行是用空格隔开的两个正整数$n$和$m$，下面的$m$行每行有三个用空格隔开的整数$i,j,b$对应着不等式$T_i-T_j\le b$。

输出格式：
如果有可行的方案，那么输出N行，每行都有一个非负整数且至少有一个为$0$，按顺序表示每个任务的起始时间。如果没有可行的方案，就输出信息NO SOLUTION。

差分约束问题，可以转为最短路来做。先求得一个可行解，接着由于需要调整最小值为$0$，所以可以先求出最小数，然后将每个数都减去那个最小数，就得到了最小数为$0$的一个解。代码如下：

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;

const int N = 1010, M = 5010;
int n, m;
int h[N], e[M], ne[M], w[M], idx;
int dist[N], cnt[N];
bool vis[N];

void add(int a, int b, int c) {
  e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}

bool spfa() {
  queue<int> q;
  for (int i = 1; i <= n; i++) {
    q.push(i);
    vis[i] = true;
  }

  while (q.size()) {
    int t = q.front(); q.pop();
    vis[t] = false;
    for (int i = h[t]; ~i; i = ne[i]) {
      int v = e[i];
      if (dist[v] > dist[t] + w[i]) {
        dist[v] = dist[t] + w[i];
        cnt[v] = cnt[t] + 1;
        if (cnt[v] > n - 1) return false;
        if (!vis[v]) {
          q.push(v);
          vis[v] = false;
        }
      }
    }
  }

  return true;
}

int main() {
  memset(h, -1, sizeof h);
  scanf("%d%d", &n, &m);

  while (m--) {
    int a, b, c;
    scanf("%d%d%d", &a, &b, &c);
    add(b, a, c);
  }

  if (!spfa()) puts("NO SOLUTION");
  else {
    int mi = 0x3f3f3f3f;
    for (int i = 1; i <= n; i++) mi = min(mi, dist[i]);
    for (int i = 1; i <= n; i++) printf("%d\n", dist[i] - mi);
  }
}

时间复杂度$O(mn)$，空间$O(n)$。