結果

問題 No.3050 Prefix Removal
ユーザー Aeren
提出日時 2025-03-07 21:30:15
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 216 ms / 2,000 ms
コード長 1,038 bytes
コンパイル時間 3,465 ms
コンパイル使用メモリ 278,244 KB
実行使用メモリ 30,620 KB
最終ジャッジ日時 2025-03-07 21:30:41
合計ジャッジ時間 11,960 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 55
権限があれば一括ダウンロードができます

ソースコード

diff #

// #include <bits/allocator.h> // Temp fix for gcc13 global pragma
// #pragma GCC target("avx2,bmi2,popcnt,lzcnt")
// #pragma GCC optimize("O3,unroll-loops")
#include <bits/stdc++.h>
// #include <x86intrin.h>
using namespace std;
#if __cplusplus >= 202002L
using namespace numbers;
#endif
#ifdef LOCAL
	#include "Debug.h"
#else
	#define debug_endl() 42
	#define debug(...) 42
	#define debug2(...) 42
	#define debugbin(...) 42
#endif



int main(){
	cin.tie(0)->sync_with_stdio(0);
	cin.exceptions(ios::badbit | ios::failbit);
	int n, k;
	cin >> n >> k;
	vector<long long> pref(n + 1);
	for(auto i = 0; i < n; ++ i){
		cin >> pref[i + 1];
		pref[i + 1] += pref[i];
	}
	multiset<long long, greater<>> s;
	long long sum = 0;
	for(auto i = 1; i < k; ++ i){
		s.insert(pref[i]);
		sum += pref[i];
	}
	long long res = numeric_limits<long long>::min() / 2;
	for(auto i = k; i <= n; ++ i){
		res = max(res, pref[i] * k - sum);
		s.insert(pref[i]);
		sum += pref[i] - *s.begin();
		s.erase(s.begin());
	}
	cout << res << "\n";
	return 0;
}

/*

*/
0