The 2021 Sichuan Provincial Collegiate Programming Contest F. Direction Setting

题面

题意

• 给定 $n$ 个点 $m$ 条边，可以任意选择边的方向，其中：
  $$D=\sum _{i=1}^n\max (0,d_i-a_i)$$
  求 $\min D$， $d_i$ 为 $i$ 的入度。并输出边方向，0 为正向，1 为负向。

• $2\le n\le 300,1\le m\le 300,0\le a_i\le 10^4,\sum n,\sum m\le 3\times 10^3$

思路

1. 几百的范围，典型的网络流。
2. 边的定向，二选一问题，以边建点，分别向两个端点连边，达到二选一效果。
3. 最小化 $D$，流量用来满足连边，最小化一个值，可以联想到最小费用。当 $d_i\le a_i$，费用为 0，此时流量免费，否则为 1。可以让点连向汇点两条边，一条容量为 $a_i$，费用为 0，一条容量为 $\inf$，费用为 1。
4. 注意：变量重名!test case 的 T 和汇点重名了。

代码

int main() {
    int t; cin >> t;
    while(t--) {
        mst(h, -1); idx = 0;
        scanf("%d %d", &n, &m);
        rep(i, 1, n) {
            scanf("%d", &a[i]);
            add(i, T, a[i], 0);
            add(i, T, inf, 1);
        }
        rep(i, 1, m) {
            int u, v; scanf("%d %d", &u, &v);
            to[i + n] = v;
            add(S, i + n, 1, 0);
            add(i + n, u, 1, 0);
            add(i + n, v, 1, 0);
        }
        printf("%d\n", ek(flow, cost));
        for(int i = 0; i < idx; i += 2) {
            int u = e[i ^ 1], v = e[i];
            if(!f[i] && n < u && v <= n) printf("%d", v != to[u]);
        }
        enter;
    }
    return 0;
}