結果

問題 No.1123 Afforestation
ユーザー 0w10w1
提出日時 2020-12-30 14:20:08
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,644 bytes
コンパイル時間 2,967 ms
コンパイル使用メモリ 222,680 KB
実行使用メモリ 102,272 KB
最終ジャッジ日時 2024-04-16 16:31:33
合計ジャッジ時間 28,673 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 WA -
testcase_44 WA -
testcase_45 WA -
testcase_46 WA -
testcase_47 WA -
testcase_48 WA -
testcase_49 WA -
testcase_50 WA -
testcase_51 WA -
testcase_52 WA -
testcase_53 WA -
testcase_54 WA -
testcase_55 WA -
testcase_56 WA -
testcase_57 WA -
testcase_58 WA -
testcase_59 WA -
testcase_60 WA -
testcase_61 WA -
testcase_62 WA -
testcase_63 WA -
testcase_64 WA -
testcase_65 WA -
testcase_66 WA -
testcase_67 WA -
testcase_68 WA -
testcase_69 WA -
testcase_70 WA -
testcase_71 WA -
testcase_72 WA -
testcase_73 WA -
testcase_74 WA -
testcase_75 WA -
testcase_76 WA -
testcase_77 WA -
testcase_78 WA -
testcase_79 WA -
testcase_80 WA -
testcase_81 WA -
testcase_82 WA -
testcase_83 WA -
testcase_84 WA -
testcase_85 WA -
testcase_86 WA -
testcase_87 WA -
testcase_88 WA -
testcase_89 WA -
testcase_90 WA -
testcase_91 WA -
testcase_92 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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 sumA = accumulate(A.begin(), A.end(), 0);
  int sumB = accumulate(B.begin(), B.end(), 0);
  cout << sumA << endl;
  int mf = ekarp.maxflow(H + W, H + W + 1);

  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";
  }

}
0