#1002539 (C++17(clang)) No.2852 Yakitori Optimization Problem

• 問題: No.2852 Yakitori Optimization Problem
• ユーザー: ryota2357
• 提出日時: 2024-08-19 21:31:07
• 言語: C++17(clang) (17.0.6 + boost 1.83.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

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	63 ms; 6,812 KB
testcase_01	AC	185 ms; 6,940 KB
testcase_02	AC	154 ms; 6,940 KB
testcase_03	AC	250 ms; 6,944 KB
testcase_04	AC	197 ms; 6,940 KB
testcase_05	AC	75 ms; 6,940 KB
testcase_06	AC	25 ms; 6,940 KB
testcase_07	AC	193 ms; 6,940 KB
testcase_08	AC	19 ms; 6,944 KB
testcase_09	AC	247 ms; 7,040 KB
testcase_10	AC	250 ms; 7,040 KB
testcase_11	AC	249 ms; 7,168 KB
testcase_12	AC	247 ms; 7,168 KB
testcase_13	AC	250 ms; 7,168 KB
testcase_14	AC	259 ms; 7,168 KB
testcase_15	AC	252 ms; 7,168 KB
testcase_16	AC	247 ms; 7,040 KB

ソースコード

#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;
}