NC50390 布局 Layout

题目链接

题目

题目描述

FJ有N头奶牛 \((2 \leq N \leq1000)\) ，编号为 \(1 \ldots N\) 。奶牛们将按照编号顺序排成一列队伍（可能有多头奶牛在同一位置上）。换句话说，假设i号奶牛位于 \(P_{ \! \;i}\) ，则 \(P_{ \,1} \leq P_{ \,2} \leq \ldots \leq P_{ \! \;N}\) 。

有些奶牛是好基友，它们希望彼此之间的距离小于等于某个数。有些奶牛是情敌，它们希望彼此之间的距离大于等于某个数。

给出 \(M_L\) ​对好基友的编号，以及它们希望彼此之间的距离小于等于多少；又给出 \(M_D\) ​对情敌的编号，以及它们希望彼此之间的距离大于等于多少 \((1 \leq M_L,\) \(M_D \leq 10^4)\) 。

请计算：如果满足上述所有条件，1号奶牛和N号奶牛之间的距离（\(P_{ \! \;N}-P_{ \,1}\)​）最大为多少。

输入描述

第一行：三个整数 \(N,M_L,M_D\) ，用空格分隔。

第 \(2 \dots M_L+1\) 行：每行三个整数A,B,D，用空格分隔，表示 \(P_B-P_A \leq D\) 。

第 \(M_L+2 \dots M_L+M_D+1\) 行：每行三个整数A,B,D，用空格分隔，表示 \(P_B-P_A \geq D\) 。

保证 \(1 \le A<B \le N,\) \(1 \leq D \leq10^6\) .

输出描述

一行，一个整数。如果没有合法方案，输出 \(\texttt{-1}\) .如果有合法方案，但1号奶牛可以与N号奶牛相距无穷远，输出 \(\texttt{-2}\) .否则，输出1号奶牛与N号奶牛间的最大距离。

示例1

输入

4 2 1
1 3 10
2 4 20
2 3 3

输出

27

说明

这四头牛分别位于0,7,10,27。

备注

对于全部数据，\(2 \leq N \leq 1000,1 \leq M_L,M_D \leq 10^4,1 \leq L,D \leq 10^6\) 。

题解

知识点：差分约束。

求最大值就求最短路，求最小值就求最长路。以最大值为例，假设 \(1 \to n\) 有路径：

\[\begin{aligned}
& x_2 - x_1 \leq w_1 \\
&\cdots \\
& x_n - x_{n-1} \leq w_{n-1}
\end{aligned}
\]

得到不等式 \(x_n-x_1 \leq \sum_{i=1}^{n-1} w_i\) ，因此 \(\max(x_n-x_1) = \min(\sum w_i)\) 即，\(x_n-x_1\) 的最大值等于 \(1\to n\) 的最短路。

这道题求最大值，因此以最短路建模，利用不等式：若有边 \(u\to v\) 且边权为 \(w\) ，则满足 \(dis[u] \leq dis[v] + w\) 。

先将问题转化：

1. 存在负环，即出现 \(a < b < c < a\) 为不合法
2. 不存在负环，但 \(1\) 不能到达 \(n\) ，则 \(1,n\) 距离可以无穷大
3. 不存在负环，\(1\) 到达 \(n\) ，则有最大值

对于负环，负环有可能和 \(1\) 不在一个连通块，导致负环没有被判断出从而有解，但实际上还是无解的，因此需要用超级源点为起点跑一次找到图中可能出现的负环。

如果没有负环，就可以从 \(1\) 跑SPFA了，后面就照常。

另外，题目隐含条件 \(x_1 \leq x_2 \leq \cdots \leq x_n\) ，需要额外建边 g.add(i+1,i,0) 不然会错。

注意两次SPFA初始化问题。

时间复杂度 \(O(n + km)\)

空间复杂度 \(O(n+m)\)

代码

#include <bits/stdc++.h>
#define ll long long

using namespace std;

template<class T>
struct Graph {
    struct edge {
        int v, nxt;
        T w;
    };
    int idx;
    vector<int> h;
    vector<edge> e;

    Graph(int n, int m):idx(0), h(n + 1), e(m + 1) {}
    void init(int n) {
        idx = 0;
        h.assign(n + 1, 0);
    }

    void add(int u, int v, T w) {
        e[++idx] = edge{ v,h[u],w };
        h[u] = idx;
    }
};

const int N = 1007, M = 20007;
Graph<int> g(N, M);
int n, ml, md;

bool vis[N];
int dis[N], cnt[N];
queue<int> q;
bool SPFA(int st) {
    for (int i = 1;i <= n + 1;i++) dis[i] = 0x3f3f3f3f, vis[i] = 0, cnt[i] = 0;
    dis[st] = 0;
    q.push(st);
    vis[st] = 1;
    while (!q.empty()) {
        int u = q.front();
        q.pop();
        vis[u] = 0;
        for (int i = g.h[u];i;i = g.e[i].nxt) {
            int v = g.e[i].v, w = g.e[i].w;
            if (dis[v] > dis[u] + w) { ///最大值跑最短路，A->B：B-A<=w（max(b-a) = min(w)），min(w)就是跑出来的结果
                dis[v] = dis[u] + w;
                cnt[v] = cnt[u] + 1;
                if (cnt[v] >= n + 1)return false;
                if (!vis[v]) {
                    q.push(v);
                    vis[v] = 1;
                }
            }
        }
    }
    return true;
}

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    cin >> n >> ml >> md;
    for (int i = 1;i <= ml;i++) {
        int u, v, w;
        cin >> u >> v >> w;
        g.add(u, v, w);
    }
    for (int i = 1;i <= md;i++) {
        int u, v, w;
        cin >> u >> v >> w;
        g.add(v, u, -w);
    }
    for (int i = 1;i <= n - 1;++i) g.add(i + 1, i, 0);
    for (int i = 1;i <= n;i++) g.add(n + 1, i, 0);
    bool ncyc = SPFA(n + 1);
    if (!ncyc) cout << -1 << '\n';///任何地方出现负环，导致方案不存在
    else {
        SPFA(1);
        if (dis[n] < 1e9)cout << dis[n] << '\n';
        else cout << -2 << '\n';
    }

    return 0;
}