Yukicoder No.177 制作進行の宮森あおいです!

,

フローに強くなりたいので解いている。 事前にフローと知らされていればさすがにやるだけな問題。

No.177 制作進行の宮森あおいです!

解説

素直な最大流。下みたいなやつ。

実装

maximum_flow()Yukicoder No.119のものと同一。

#include <iostream>
#include <vector>
#include <set>
#include <queue>
#define repeat(i,n) for (int i = 0; (i) < (n); ++(i))
typedef long long ll;
using namespace std;
ll maximum_flow(int s, int t, vector<vector<ll> > const & g /* capacity, adjacency matrix */) { // edmonds karp, O(E^2V)
    int n = g.size();
    vector<vector<ll> > flow(n, vector<ll>(n));
    auto residue = [&](int i, int j) { return g[i][j] - flow[i][j]; };
    ll result = 0;
    while (true) {
        vector<int> prev(n, -1);
        vector<ll> f(n);
        // find the shortest augmenting path
        queue<int> q; // bfs
        q.push(s);
        while (not q.empty()) {
            int i = q.front(); q.pop();
            repeat (j,n) if (prev[j] == -1 and j != s and residue(i,j) > 0) {
                prev[j] = i;
                f[j] = residue(i,j);
                if (i != s) f[j] = min(f[j], f[i]);
                q.push(j);
            }
        }
        if (prev[t] == -1) break; // not found
        // backtrack
        for (int i = t; prev[i] != -1; i = prev[i]) {
            int j = prev[i];
            flow[j][i] += f[t];
            flow[i][j] -= f[t];
        }
        result += f[t];
    }
    return result;
}
int main() {
    // input
    int w, n; cin >> w >> n;
    vector<int> j(n); repeat (i,n) cin >> j[i];
    int m; cin >> m;
    vector<int> c(m); repeat (i,m) cin >> c[i];
    vector<set<int> > x(m);
    repeat (i,m) {
        int q; cin >> q;
        repeat (j,q) {
            int y; cin >> y;
            x[i].insert(y-1);
        }
    }
    // make a capacity graph
    const ll INF = 1000000007;
    vector<vector<ll> > g(n+m+2, vector<ll>(n+m+2));
    int s = n+m, t = n+m+1;
    repeat (i,n) g[s][i] = j[i];
    repeat (j,m) g[n+j][t] = c[j];
    repeat (j,m) repeat (i,n) if (not x[j].count(i)) g[i][n+j] = INF;
    // output
    cout << (maximum_flow(s,t,g) >= w ? "SHIROBAKO" : "BANSAKUTSUKITA") << endl;
    return 0;
}