結果

問題 No.3050 Prefix Removal
ユーザー 沙耶花
提出日時 2025-03-07 21:24:00
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 258 ms / 2,000 ms
コード長 697 bytes
コンパイル時間 4,479 ms
コンパイル使用メモリ 252,308 KB
実行使用メモリ 13,212 KB
最終ジャッジ日時 2025-03-07 21:24:17
合計ジャッジ時間 15,484 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 55
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000000
#define Inf64 1000000000000000001LL

int main(){
	
	int N,K;
	cin>>N>>K;
	vector<long long> a(N);
	rep(i,N){
		cin>>a[i];
	}
	long long ans = -Inf64*4;
	
	priority_queue<long long,vector<long long>,greater<long long>> Q;
	long long qs = 0;
	long long sum = 0;
	rep(i,N){
		sum += a[i];
		if(i!=0){
			Q.push(a[i] - sum);
			qs += a[i]-sum;
		}
		while(Q.size()>K-1){
			qs -= Q.top();
			Q.pop();
		}
		if(Q.size()==K-1)ans = max(ans,qs + sum * K);
	}
	cout<<ans<<endl;
	return 0;
}
0