結果

問題 No.1117 数列分割
ユーザー snrnsidy
提出日時 2021-06-10 03:25:10
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,217 bytes
コンパイル時間 2,133 ms
コンパイル使用メモリ 196,320 KB
最終ジャッジ日時 2025-01-22 05:08:20
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

long long int dp[2][3001];
long long int psum[3001];
long long int C[3001][3001];
vector <long long int> v;
int N, K, M;

void solve(int t, int l, int r, int s, int e)
{
	if (l > r)
	{
		return;
	}
	int m = (l + r) / 2;
	dp[1][m] = dp[0][s - 1] + C[s][m];
	int opt = s;
	int end = min(e, m);
	for (int k = s + 1; k <= end; k++)
	{
		if (m - k + 1 > M)
		{
			break;
		}
		long long int val = dp[0][k - 1] + C[k][m];
		if (dp[1][m] < val)
		{
			dp[1][m] = val;
			opt = k;
		}
	}
	if (l != r)
	{
		solve(t, l, m - 1, s, opt);
		solve(t, m + 1, r, opt, e);
	}
}
int main(void)
{
	cin.tie(0);
	ios::sync_with_stdio(false);
	
	cin >> N >> K >> M;
	for (int i = 0; i < N; i++)
	{
		int t;
		cin >> t;
		v.push_back(t);
	}

	for (int i = 1; i <= N; i++)
	{
		psum[i] = v[i - 1];
	}

	for (int i = 1; i <= N; i++)
	{
		psum[i] += psum[i - 1];
	}

	for (int i = 1; i <= N; i++)
	{
		for (int j = 1; j <= N; j++)
		{
			C[i][j] = abs(psum[j] - psum[i - 1]);
		}
	}

	for (int i = 1; i <= N; i++)
	{
		dp[1][i] = -1e18;
	}
	dp[1][0] = 0;
	for (int i = 1; i <= K; i++)
	{
		memcpy(dp[0], dp[1], sizeof(dp[1]));
		solve(i, i, N, i, N);
	}

	cout << dp[1][N] << '\n';

	return 0;
}
0