結果

問題 No.2549 Paint Eggs
コンテスト
ユーザー InTheBloom
提出日時 2023-11-20 19:51:53
言語 D
(dmd 2.109.1)
結果
WA  
実行時間 -
コード長 1,194 bytes
コンパイル時間 2,357 ms
コンパイル使用メモリ 174,720 KB
実行使用メモリ 23,964 KB
最終ジャッジ日時 2024-09-26 06:41:49
合計ジャッジ時間 8,012 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 29 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std;

void main () {
    int N, M, K; readln.read(N, M, K);
    int[] C = readln.split.to!(int[]);
    C[] -= 1;
    int[] A = readln.split.to!(int[]);

    solve(N, M, K, C, A);
}

void solve (int N, int M, int K, int[] C, int[] A) {
    /* スライドしていけば良さそう */
    int[int] mp;
    // 最初の一回
    foreach (i; 0..K) mp[C[i]]++;

    long ans = long.max;

    foreach (key, val; mp) {
        ans = min(ans, 1L*(K-val)*A[key]);
    }

    int left = 0, right = K;
    while (right < N) {
        // [left, right)を保持している
        // leftを除去してrightを加入させる

        mp[C[left]]--;
        mp[C[right]]++;
        ans = min(ans, 1L*(K-mp[C[left]])*A[C[left]]);
        ans = min(ans, 1L*(K-mp[C[right]])*A[C[right]]);

        left++, right++;

        //dbg
        //writeln("[", left, ", ", right, ") ", mp);
    }

    // ハマった。単に全部塗り替えるのが最適な場合がある
    foreach (c; C) {
        ans = min(ans, 1L*K*A[c]);
    }

    writeln(ans);
}

void read(T...)(string S, ref T args) {
    auto buf = S.split;
    foreach (i, ref arg; args) {
        arg = buf[i].to!(typeof(arg));
    }
}
0