結果

問題 No.1028 闇討ち
ユーザー Strorkis
提出日時 2020-04-18 01:20:31
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 231 ms / 2,000 ms
コード長 1,199 bytes
コンパイル時間 885 ms
コンパイル使用メモリ 73,580 KB
実行使用メモリ 15,360 KB
最終ジャッジ日時 2024-10-03 20:14:55
合計ジャッジ時間 3,873 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>

int main() {
    int N;
    std::cin >> N;

    std::vector<std::vector<int>> A(N, std::vector<int>(N));
    std::vector<int> b(N), res(N);
    for (int i = 0; i < N; i++)
        for (int j = 0; j < N; j++) {
            std::cin >> A[i][j];
            A[i][j]--;
            b[A[i][j]] += std::max(i, j);
            res[A[i][j]] += std::max(i, j);
        }

    std::vector<std::vector<int>> cs1(N, std::vector<int>(N)),
                                  cs2(N, std::vector<int>(N));
    for (int k = 0; k < N - 1; k++)
        for (int i = 0; i <= k; i++) {
            cs1[k][A[i][k - i]]++;
            cs2[N - 2 - k][A[N - 1 - i][k - i]]++;
        }
    for (int k = 1; k < N - 1; k++)
        for (int i = 0; i < N; i++) {
            cs1[k][i] += cs1[k - 1][i];
            cs2[N - 2 - k][i] += cs2[N - 2 - k + 1][i];
        }

    for (int k = 0; k < N - 1; k++) {
        for (int i = 0; i < N; i++)
            b[i] += cs1[k][i] - cs2[k][i];

        for (int i = 0; i < N; i++)
            res[i] = std::min(res[i], b[i]);
    }
          
    int ans = 0;
    for (int i = 0; i < N; i++)
        ans += res[i];
    std::cout << ans << "\n";
}
0