結果

問題 No.1117 数列分割
ユーザー kotatsugame
提出日時 2020-07-17 21:54:17
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,074 ms / 3,000 ms
コード長 778 bytes
コンパイル時間 827 ms
コンパイル使用メモリ 78,348 KB
実行使用メモリ 74,624 KB
最終ジャッジ日時 2024-11-30 03:23:00
合計ジャッジ時間 9,433 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:8:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
    8 | main()
      | ^~~~

ソースコード

diff #

#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
int N,K,M;
int A[3030];
long dp[3030][3030];
main()
{
	cin>>N>>K>>M;
	for(int i=0;i<N;i++)cin>>A[i];
	for(int i=1;i<=N;i++)dp[0][i]=-1e18;
	for(int i=0;i<K;i++)
	{
		{
			priority_queue<pair<long,int> >P;
			long cumsum=0;
			for(int j=1;j<=N;j++)
			{
				P.push(make_pair(dp[i][j-1]-cumsum,j-1));
				cumsum+=A[j-1];
				while(P.top().second<j-M)P.pop();
				dp[i+1][j]=max(dp[i+1][j],P.top().first+cumsum);
			}
		}
		{
			priority_queue<pair<long,int> >P;
			long cumsum=0;
			for(int j=1;j<=N;j++)
			{
				P.push(make_pair(dp[i][j-1]-cumsum,j-1));
				cumsum-=A[j-1];
				while(P.top().second<j-M)P.pop();
				dp[i+1][j]=max(dp[i+1][j],P.top().first+cumsum);
			}
		}
	}
	cout<<dp[K][N]<<endl;
}
0