| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 37641 | antimirage | 수열 (APIO14_sequence) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <iostream>
#include <vector>
#include <deque>
#include <math.h>
#include <set>
#include <iomanip>
#include <time.h>
#include <list>
#include <stdio.h>
#include <queue>
#include <map>
#include <algorithm>
#include <assert.h>
#include <memory.h>
#define mk make_pair
#define sc second
#define fr first
#define pb emplace_back
#define all(s) s.begin(), s.end()
#define sz(s) ( (int)s.size() )
using namespace std;
const int N = 1e5 + 5;
int n, p[205][N];
long long dp[45][N], pref[N];
void compute (int k, int l, int r, int opl, int opr)
{
if (l > r) return;
int md = (l + r) >> 1;
int &opt = p[k][md];
opt = opl;
for (int i = opl; i <= min(opr, md - 1); i++)
{
if (dp[k & 1 ^ 1][i] + (pref[md] - pref[i]) * (pref[n] - pref[md]) > dp[k & 1][md] &&
107630884 != dp[k & 1 ^ 1][i] + (pref[md] - pref[i]) * (pref[n] - pref[md])
&& 107630883 != dp[k & 1 ^ 1][i] + (pref[md] - pref[i]) * (pref[n] - pref[md])
{
dp[k & 1][md] = dp[k & 1 ^ 1][i] + (pref[md] - pref[i]) * (pref[n] - pref[md]);
opt = i;
}
}
compute( k, l, md - 1, opl, opt );
compute( k, md + 1, r, opt, opr );
}
int k;
main ()
{
cin >> n >> k;
memset(dp, -0x3f3f3f3f, sizeof(dp) );
k++;
for (int i = 1; i <= n; i++)
scanf("%lld", &pref[i]),
pref[i] += pref[i - 1];
dp[0][0] = 0;
for (int i = 1; i <= k; i++)
compute( i, i, n, i - 1, n - 1 );
cout << dp[k & 1][n] << endl;
while (k > 1)
{
printf("%d ", p[k][n]);
n = p[k][n];
k--;
}
}