結果

問題 No.2549 Paint Eggs
コンテスト
ユーザー inkyaman
提出日時 2024-03-24 15:06:44
言語 C++17
(gcc 15.2.0 + boost 1.90.0)
コンパイル:
g++-15 -O2 -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 514 ms / 2,000 ms
コード長 1,293 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 959 ms
コンパイル使用メモリ 146,792 KB
実行使用メモリ 19,968 KB
最終ジャッジ日時 2026-07-04 06:32:43
合計ジャッジ時間 18,408 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 45
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#define _USE_MATH_DEFINES
#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <queue>
#include <math.h>
#include <cmath>
#include <stack>
#include <map>
#include <set>
#include <numeric>
#include <iomanip>
#include <climits>
#include <functional>
#include <cassert>
#include <tuple>
using namespace std;
using ll = long long;

int N, M, K;
int C[202020];
ll A[202020], S[202020];

int main() {

    cin >> N >> M >> K;
    for(int i = 1; i <= N; ++i) cin >> C[i];
    for(int i = 1; i <= M; ++i) cin >> A[i];
    
    multiset<pair<ll,int>> mst;
    for(int i = 1; i <= M; ++i) mst.insert({K*A[i],i});
    vector<ll> cost(M+1, 0);
    for(int i = 1; i <= M; ++i) cost[i] = K*A[i];

    for(int i = 1; i <= K-1; ++i) {
        mst.erase({cost[C[i]], C[i]});
        cost[C[i]] -= A[C[i]];
        mst.insert({cost[C[i]], C[i]});
    }

    ll ans = 1e18;
    for(int i = K; i <= N; ++i) {
        mst.erase({cost[C[i]], C[i]});
        cost[C[i]] -= A[C[i]];
        mst.insert({cost[C[i]], C[i]});

        auto itr = mst.begin();
        ll mn = (*itr).first;
        ans = min(ans, mn);

        mst.erase({cost[C[i-K+1]], C[i-K+1]});
        cost[C[i-K+1]] += A[C[i-K+1]];
        mst.insert({cost[C[i-K+1]], C[i-K+1]});

    }

    cout << ans << endl;

}
0