結果

問題 No.2852 Yakitori Optimization Problem
ユーザー ryota2357ryota2357
提出日時 2024-08-19 21:31:07
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
AC  
実行時間 259 ms / 2,000 ms
コード長 863 bytes
コンパイル時間 1,500 ms
コンパイル使用メモリ 132,276 KB
実行使用メモリ 7,168 KB
最終ジャッジ日時 2024-08-19 21:31:13
合計ジャッジ時間 6,200 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <cassert>
#include <vector>
#include <algorithm>
using namespace std;
using ll = long long;

int main() {
	int n, k;
	cin >> n >> k;
	vector<int> a(n), b(n), c(n);
	for (int i = 0; i < n; ++i) cin >> a[i];
	for (int i = 0; i < n; ++i) cin >> b[i];
	for (int i = 0; i < n; ++i) cin >> c[i];
	
	assert(1 <= n && n <= 200000);
	assert(0 <= k && k <= n);
	for (int i = 0; i < n; ++i) {
		assert(1 <= a[i] && a[i] <= 1000'000'000);
		assert(1 <= b[i] && b[i] <= 1000'000'000);
		assert(1 <= c[i] && c[i] <= 1000'000'000);
	}
	
	ll sum = 0;
	for (int i = 0; i < n; ++i) sum += a[i];
	vector<pair<int, int>> bc(n);
	for (int i = 0; i < n; ++i) bc[i] = {b[i] - c[i], i};
	sort(bc.rbegin(), bc.rend());
	for (int cnt = 0; cnt < n; ++cnt) {
		auto [_, i] = bc[cnt];
		if (cnt < k) sum += b[i];
		else sum += c[i];
	}
	cout << sum << endl;
}
0