結果

問題 No.1123 Afforestation
ユーザー e869120
提出日時 2020-07-02 23:59:29
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 712 ms / 2,500 ms
コード長 3,376 bytes
コンパイル時間 1,213 ms
コンパイル使用メモリ 85,532 KB
実行使用メモリ 124,672 KB
最終ジャッジ日時 2024-09-16 17:04:03
合計ジャッジ時間 15,509 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 90
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
#pragma warning (disable: 4996)
struct Node {
    int to, cap, rev;
};
class MaxFlow {
public:
    vector<Node> G[5009];
    bool used[5009];
    void add_edge(int u, int v, int c, int d) {
        G[u].push_back(Node{ v, c - d, (int)G[v].size() });
        G[v].push_back(Node{ u, d, (int)G[u].size() - 1 });
    }
    int dfs(int pos, int to, int f) {
        if (pos == to) return f;
        used[pos] = true;
        for (int i = 0; i < G[pos].size(); i++) {
            if (used[G[pos][i].to] == true || G[pos][i].cap == 0) continue;
            int F = dfs(G[pos][i].to, to, min(f, G[pos][i].cap));
            if (F == 0) continue;
            G[pos][i].cap -= F;
            G[G[pos][i].to][G[pos][i].rev].cap += F;
            return F;
        }
        return 0;
    }
    int max_flow(int u, int v) {
        int f = 0;
        while (true) {
            for (int i = 0; i < 5009; i++) used[i] = false;
            int g = dfs(u, v, 100000007); f += g;
            if (g == 0) break;
        }
        return f;
    }
};
// 入力
int H, A[2509];
int W, B[2509];
int K, X[19], Y[19];
// その他
int Answer[2509][2509];
bool used[2509][2509];
bool forced[2509][2509];
MaxFlow G;
int main() {
    // 入力
    cin >> H >> W;
    for (int i = 1; i <= H; i++) cin >> A[i];
    for (int i = 1; i <= W; i++) cin >> B[i];
    cin >> K;
    for (int i = 1; i <= K; i++) cin >> X[i] >> Y[i];
    // 貪欲法
    vector<pair<int, int>> CA, CB;
    for (int i = 1; i <= K; i++) forced[X[i]][Y[i]] = true;
    for (int i = 1; i <= H; i++) CA.push_back(make_pair(A[i], i));
    for (int i = 1; i <= W; i++) CB.push_back(make_pair(B[i], i));
    sort(CA.begin(), CA.end()); reverse(CA.begin(), CA.end());
    sort(CB.begin(), CB.end()); reverse(CB.begin(), CB.end());
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < CA[i].first; j++) used[CA[i].second][CB[j].second] = true;
        for (int j = 0; j < CA[i].first; j++) CB[j].first -= 1;
        sort(CB.begin(), CB.end()); reverse(CB.begin(), CB.end());
    }
    if (CB[0].first >= 1) {
        printf(":(\n");
        return 0;
    }
    // 判定
    int SA = 0; for (int i = 1; i <= H; i++) SA += A[i];
    int SB = 0; for (int i = 1; i <= W; i++) SB += B[i];
    if (SA != SB) {
        printf(":(\n");
        return 0;
    }
    // 家の部分を消去
    int cnts = 0;
    for (int i = 1; i <= K; i++) {
        if (used[X[i]][Y[i]] == false) continue;
        used[X[i]][Y[i]] = false; cnts++;
    }
    // 辺を貼る
    for (int i = 1; i <= H; i++) {
        for (int j = 1; j <= W; j++) {
            if (forced[i][j] == true) continue;
            G.add_edge(i, H + j, 1, (int)used[i][j]);
        }
    }
    for (int i = 1; i <= H; i++) {
        int cnts = 0;
        for (int j = 1; j <= W; j++) cnts += (int)used[i][j];
        G.add_edge(H + W + 1, i, A[i], cnts);
    }
    for (int i = 1; i <= W; i++) {
        int cnts = 0;
        for (int j = 1; j <= H; j++) cnts += used[j][i];
        G.add_edge(H + i, H + W + 2, B[i], cnts);
    }
    // フローを流す
    int Ret = G.max_flow(H + W + 1, H + W + 2);
    if (Ret != cnts) {
        printf(":(\n");
        return 0;
    }
    // 復元する
    for (int i = 1; i <= H; i++) {
        for (int j = 0; j < G.G[i].size(); j++) {
            int pos = G.G[i][j].to; if (pos <= H || pos >= H + W + 1) continue;
            if (G.G[i][j].cap == 0) Answer[i][pos - H] = 1;
        }
    }
    for (int i = 1; i <= K; i++) Answer[X[i]][Y[i]] = 2;
    // 出力
    printf("Yay!\n");
    for (int i = 1; i <= H; i++) {
        for (int j = 1; j <= W; j++) {
            if (Answer[i][j] == 0) printf(".");
            if (Answer[i][j] == 1) printf("o");
            if (Answer[i][j] == 2) printf("x");
        }
        printf("\n");
    }
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0