結果

問題 No.1225 I hate I hate Matrix Construction
コンテスト
ユーザー ooaiu
提出日時 2024-12-11 11:39:38
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 5 ms / 2,000 ms
コード長 1,926 bytes
コンパイル時間 3,645 ms
コンパイル使用メモリ 250,864 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-11 11:39:43
合計ジャッジ時間 4,809 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#ifndef LOCAL
#include <bits/stdc++.h>

using namespace std;

#define debug(...) (void(0))
#else
#include "algo/debug.h"
#endif

void solve() {
    int N;
    cin >> N;
    vector<int> S(N), T(N);
    for (int i = 0; i < N; i++) cin >> S[i];
    for (int i = 0; i < N; i++) cin >> T[i];
    vector<vector<int>> ans(N, vector<int>(N, 0));
    for (int i = 0; i < N; i++) {
        if (S[i] == 2)
            for (int j = 0; j < N; j++) ans[i][j] = 1;
        if (T[i] == 2)
            for (int j = 0; j < N; j++) ans[j][i] = 1;
    }
    vector<vector<int>> count(N, vector<int>(N));
    for (int i = 0; i < N; i++)
        if (S[i] == 1) {
            bool f = false;
            for (int j = 0; j < N; j++) {
                if (ans[i][j] == 1) f = true;
            }
            if (!f) {
                for (int j = 0; j < N; j++) count[i][j]++;
            }
        }
    for (int i = 0; i < N; i++)
        if (T[i] == 1) {
            bool f = false;
            for (int j = 0; j < N; j++) {
                if (ans[j][i] == 1) f = true;
            }
            if (!f) {
                for (int j = 0; j < N; j++) count[j][i]++;
            }
        }
    const auto Dec = [&](int i, int j) -> void {
        for (int k = 0; k < N; k++) count[i][k]--;
        for (int k = 0; k < N; k++) count[k][j]--;
    };
    for (int z : {2, 1}) {
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if (count[i][j] == z) {
                    ans[i][j] = 1;
                    Dec(i, j);
                }
            }
        }
    }
    debug(ans);
    int sum = 0;
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            sum += ans[i][j];
        }
    }
    cout << sum << endl;
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int tt = 1;
    // std::cin >> tt;
    while (tt--) {
        solve();
    }
}
0