#598362 (C++17) No.1123 Afforestation

提出ソース

結果

問題	No.1123 Afforestation
ユーザー	0w1
提出日時	2020-12-30 14:20:43
言語	C++17 (gcc 13.2.0 + boost 1.83.0)
結果	AC
実行時間	499 ms / 2,500 ms
コード長	3,620 bytes
コンパイル時間	3,082 ms
コンパイル使用メモリ	221,564 KB
実行使用メモリ	102,272 KB
最終ジャッジ日時	2024-04-16 16:31:50
合計ジャッジ時間	15,682 ms
ジャッジサーバーID （参考情報）	judge4 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	8 ms 6,144 KB
testcase_05	AC	86 ms 28,416 KB
testcase_06	AC	477 ms 102,272 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	5 ms 5,376 KB
testcase_12	AC	34 ms 14,720 KB
testcase_13	AC	493 ms 102,144 KB
testcase_14	AC	2 ms 5,376 KB
testcase_15	AC	2 ms 5,376 KB
testcase_16	AC	2 ms 5,376 KB
testcase_17	AC	2 ms 5,376 KB
testcase_18	AC	3 ms 5,376 KB
testcase_19	AC	10 ms 6,144 KB
testcase_20	AC	93 ms 28,160 KB
testcase_21	AC	481 ms 102,144 KB
testcase_22	AC	497 ms 101,888 KB
testcase_23	AC	499 ms 102,144 KB
testcase_24	AC	490 ms 102,144 KB
testcase_25	AC	489 ms 102,144 KB
testcase_26	AC	2 ms 5,376 KB
testcase_27	AC	2 ms 5,376 KB
testcase_28	AC	400 ms 98,304 KB
testcase_29	AC	2 ms 5,376 KB
testcase_30	AC	2 ms 5,376 KB
testcase_31	AC	441 ms 98,432 KB
testcase_32	AC	2 ms 5,376 KB
testcase_33	AC	2 ms 5,376 KB
testcase_34	AC	2 ms 5,376 KB
testcase_35	AC	3 ms 5,376 KB
testcase_36	AC	2 ms 5,376 KB
testcase_37	AC	3 ms 5,376 KB
testcase_38	AC	2 ms 5,376 KB
testcase_39	AC	3 ms 5,376 KB
testcase_40	AC	2 ms 5,376 KB
testcase_41	AC	3 ms 5,376 KB
testcase_42	AC	6 ms 5,376 KB
testcase_43	AC	10 ms 6,016 KB
testcase_44	AC	19 ms 8,576 KB
testcase_45	AC	18 ms 8,576 KB
testcase_46	AC	23 ms 9,600 KB
testcase_47	AC	23 ms 9,472 KB
testcase_48	AC	2 ms 5,376 KB
testcase_49	AC	2 ms 5,376 KB
testcase_50	AC	3 ms 5,376 KB
testcase_51	AC	3 ms 5,376 KB
testcase_52	AC	2 ms 5,376 KB
testcase_53	AC	2 ms 5,376 KB
testcase_54	AC	2 ms 5,376 KB
testcase_55	AC	3 ms 5,376 KB
testcase_56	AC	3 ms 5,376 KB
testcase_57	AC	3 ms 5,376 KB
testcase_58	AC	5 ms 5,376 KB
testcase_59	AC	9 ms 6,016 KB
testcase_60	AC	19 ms 8,576 KB
testcase_61	AC	19 ms 8,576 KB
testcase_62	AC	24 ms 9,472 KB
testcase_63	AC	23 ms 9,600 KB
testcase_64	AC	2 ms 5,376 KB
testcase_65	AC	2 ms 5,376 KB
testcase_66	AC	2 ms 5,376 KB
testcase_67	AC	2 ms 5,376 KB
testcase_68	AC	3 ms 5,376 KB
testcase_69	AC	4 ms 5,376 KB
testcase_70	AC	6 ms 5,376 KB
testcase_71	AC	10 ms 6,144 KB
testcase_72	AC	2 ms 5,376 KB
testcase_73	AC	2 ms 5,376 KB
testcase_74	AC	2 ms 5,376 KB
testcase_75	AC	2 ms 5,376 KB
testcase_76	AC	2 ms 5,376 KB
testcase_77	AC	2 ms 5,376 KB
testcase_78	AC	2 ms 5,376 KB
testcase_79	AC	2 ms 5,376 KB
testcase_80	AC	2 ms 5,376 KB
testcase_81	AC	1 ms 5,376 KB
testcase_82	AC	2 ms 5,376 KB
testcase_83	AC	2 ms 5,376 KB
testcase_84	AC	329 ms 102,144 KB
testcase_85	AC	352 ms 102,272 KB
testcase_86	AC	2 ms 5,376 KB
testcase_87	AC	2 ms 5,376 KB
testcase_88	AC	224 ms 102,144 KB
testcase_89	AC	254 ms 102,272 KB
testcase_90	AC	2 ms 5,376 KB
testcase_91	AC	2 ms 5,376 KB
testcase_92	AC	2 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;

template<typename T = int, T INF = 0x3f3f3f3f>
struct EKarp {
  int n;
  vector<vector<T>> capacity;
  vector<vector<int>> adj;

  EKarp(int _n) : n(_n), capacity(_n, vector<T>(_n)), adj(_n) {}

  void addEdge(T w, int u, int v) {
    adj[u].push_back(v);
    adj[v].push_back(u);
    capacity[u][v] += w;
  }

  T bfs(int s, int t, vector<int> &parent) {
    fill(parent.begin(), parent.end(), -1);
    parent[s] = -2;
    queue<pair<int, int>> que;
    que.push({ s, INF });
    while (que.size()) {
      auto [cur, flow] = que.front();
      que.pop();
      for (int nxt : adj[cur]) {
        if (parent[nxt] == -1 && capacity[cur][nxt]) {
          parent[nxt] = cur;
          T nflow = min(flow, capacity[cur][nxt]);
          if (nxt == t) return nflow;
          que.push({ nxt, nflow });
        }
      }
    }
    return (T) 0;
  }

  T maxflow(int s, int t) {
    T flow = 0;
    vector<int> parent(n);
    T nflow;
    while (nflow = bfs(s, t, parent)) {
      flow += nflow;
      int cur = t;
      while (cur != s) {
        int pre = parent[cur];
        capacity[pre][cur] -= nflow;
        capacity[cur][pre] += nflow;
        cur = pre;
      }
    }
    return flow;
  }
};

int main() {
  ios::sync_with_stdio(false);

  int H, W; {
    cin >> H >> W;
  }

  vector<int> A(H); {
    for (int i = 0; i < H; ++i) cin >> A[i];
  }

  vector<int> B(W); {
    for (int i = 0; i < W; ++i) cin >> B[i];
  }

  int K; {
    cin >> K;
  }

  vector<int> X(K), Y(K); {
    for (int i = 0; i < K; ++i) cin >> X[i] >> Y[i], --X[i], --Y[i];
  }

  set<pair<int, int>> badBag; {
    for (int i = 0; i < K; ++i) badBag.emplace(X[i], Y[i]);
  }

  EKarp ekarp(H + W + 2); {
    for (int i = 0; i < H; ++i) ekarp.addEdge(A[i], H + W, i);
    for (int i = 0; i < W; ++i) ekarp.addEdge(B[i], H + i, H + W + 1);
    for (int i = 0; i < H; ++i) {
      for (int j = 0; j < W; ++j) {
        int w = !badBag.count(make_pair(i, j));
        ekarp.addEdge(w, i, H + j);
      }
    }
  }

  int gc = 0; {
    auto a = A;
    auto b = B;

    vector<int> aord(H); {
      iota(aord.begin(), aord.end(), 0);
      sort(aord.begin(), aord.end(), [&](int i, int j) { return a[i] > a[j]; });
    }

    for (int i : aord) {
      vector<int> bord(W); {
        iota(bord.begin(), bord.end(), 0);
        sort(bord.begin(), bord.end(), [&](int i, int j) { return b[i] > b[j]; });
      }

      for (int j : bord) {
        if (badBag.count(make_pair(i, j))) continue;
        if (a[i] && b[j]) {
          --a[i];
          --b[j];
          assert(ekarp.capacity[i][H + j] == 1);
          ekarp.capacity[i][H + j] -= 1;
          ekarp.capacity[H + j][i] += 1;
          assert(ekarp.capacity[H + W][i] > 0);
          ekarp.capacity[H + W][i] -= 1;
          ekarp.capacity[i][H + W] += 1;
          assert(ekarp.capacity[H + j][H + W + 1] > 0);
          ekarp.capacity[H + j][H + W + 1] -= 1;
          ekarp.capacity[H + W + 1][H + j] += 1;
          ++gc;
        }
      }
    }
  }

  int mf = ekarp.maxflow(H + W, H + W + 1);
  int sumA = accumulate(A.begin(), A.end(), 0);
  int sumB = accumulate(B.begin(), B.end(), 0);

  if (gc + mf != max(sumA, sumB)) {
    cout << ":(" << endl;
  } else {
    cout << "Yay!" << endl;
    vector<string> res(H, string(W, '.')); {
      for (int i = 0; i < K; ++i) {
        res[X[i]][Y[i]] = 'x';
      }
      for (int i = 0; i < H; ++i) {
        for (int j = 0; j < W; ++j) {
          if (ekarp.capacity[H + j][i] == 1) res[i][j] = 'o';
        }
      }
    }
    for (int i = 0; i < H; ++i) cout << res[i] << "\n";
  }

}

yukicoder

結果

テストケース

ソースコード