結果

問題 No.1117 数列分割
ユーザー kotatsugamekotatsugame
提出日時 2020-07-17 21:49:39
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 1,822 bytes
コンパイル時間 1,450 ms
コンパイル使用メモリ 88,292 KB
実行使用メモリ 53,788 KB
最終ジャッジ日時 2024-05-07 11:18:13
合計ジャッジ時間 17,365 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
13,892 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 528 ms
19,108 KB
testcase_04 AC 502 ms
18,248 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 6 ms
6,940 KB
testcase_07 AC 5 ms
6,940 KB
testcase_08 AC 371 ms
14,312 KB
testcase_09 AC 123 ms
7,888 KB
testcase_10 AC 335 ms
14,048 KB
testcase_11 AC 1,679 ms
29,024 KB
testcase_12 AC 2,508 ms
32,016 KB
testcase_13 AC 2,322 ms
47,792 KB
testcase_14 AC 2,892 ms
38,820 KB
testcase_15 TLE -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:68:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   68 | main()
      | ^~~~

ソースコード

diff #

#include<iostream>
#include<algorithm>
using namespace std;
#include<vector>
#include<functional>
template<typename T>
struct lazysegtree{
	int n;
	const T defvalue=(long)-1e18,lazydefvalue=0L;
	vector<T>dat,lazy;
	lazysegtree(int n_=0)
	{
		n=1;
		while(n<n_)n<<=1;
		dat.assign(2*n-1,defvalue);
		lazy.assign(2*n-1,lazydefvalue);
	}
	void copy(const vector<T>&v)
	{
		for(int i=0;i<v.size();i++)dat[i+n-1]=v[i];
		for(int i=n-2;i>=0;i--)dat[i]=max(dat[2*i+1],dat[2*i+2]);
	}
	void eval(int i,int l,int r)
	{
		if(lazy[i])
		{
			dat[i]+=lazy[i];
			if(r-l>1)
			{
				lazy[2*i+1]+=lazy[i];
				lazy[2*i+2]+=lazy[i];
			}
			lazy[i]=lazydefvalue;
		}
	}
	void update(int a,int b,T x,int k=0,int l=0,int r=-1)//[a,b)
	{
		if(r<0)r=n;
		eval(k,l,r);
		if(b<=l||r<=a)return;
		else if(a<=l&&r<=b)
		{
			lazy[k]+=x;
			eval(k,l,r);
		}
		else
		{
			update(a,b,x,2*k+1,l,(l+r)/2);
			update(a,b,x,2*k+2,(l+r)/2,r);
			dat[k]=max(dat[2*k+1],dat[2*k+2]);
		}
	}
	T query(int a,int b,int k=0,int l=0,int r=-1)//[a,b)
	{
		if(r<0)r=n;
		eval(k,l,r);
		if(b<=l||r<=a)return defvalue;
		else if(a<=l&&r<=b)return dat[k];
		else return max(
			query(a,b,2*k+1,l,(l+r)/2),
			query(a,b,2*k+2,(l+r)/2,r)
		);
	}
};
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++)
	{
		vector<long>init(N+1);
		for(int j=0;j<=N;j++)
		{
			init[j]=dp[i][j];
			dp[i+1][j]=-1e18;
		}
		{
			lazysegtree<long>P(N+1);
			P.copy(init);
			for(int j=1;j<=N;j++)
			{
				P.update(j-M,j,A[j-1]);
				dp[i+1][j]=max(dp[i+1][j],P.query(j-M,j));
			}
		}
		{
			lazysegtree<long>P(N+1);
			P.copy(init);
			for(int j=1;j<=N;j++)
			{
				P.update(j-M,j,-A[j-1]);
				dp[i+1][j]=max(dp[i+1][j],P.query(j-M,j));
			}
		}
	}
	cout<<dp[K][N]<<endl;
}
0