結果

問題 No.2549 Paint Eggs
ユーザー Today03
提出日時 2023-11-25 18:12:16
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 246 ms / 2,000 ms
コード長 975 bytes
コンパイル時間 1,973 ms
コンパイル使用メモリ 199,440 KB
最終ジャッジ日時 2025-02-18 01:04:49
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 45
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#if __has_include("Today's/debug.cpp")
#include "Today's/debug.cpp"
#else
#define debug(...)
#define print_line
#define debugc(...)
#define cerr \
	if (false) cerr
#endif
#include <atcoder/segtree>
using namespace atcoder;

long long op(long long a, long long b) {
	return min(a, b);
}
long long e() {
	return LLONG_MAX;
}

int main() {
	int N, M, K;
	cin >> N >> M >> K;
	vector<int> C(N);
	for (int i = 0; i < N; i++) {
		cin >> C[i];
		C[i]--;
	}
	vector<long long> A(M);
	for (int i = 0; i < M; i++) {
		cin >> A[i];
	}
	segtree<long long, op, e> st(M);
	for (int i = 0; i < M; i++) {
		st.set(i, A[i] * K);
	}
	long long ans = LLONG_MAX;
	for (int i = 0; i < K; i++) {
		st.set(C[i], st.get(C[i]) - A[C[i]]);
	}
	for (int i = 0; i + K <= N; i++) {
		ans = min(ans, st.all_prod());
		st.set(C[i], st.get(C[i]) + A[C[i]]);
		if (i + K < N) {
			st.set(C[i + K], st.get(C[i + K]) - A[C[i + K]]);
		}
	}
	cout << ans << endl;
}
0