Pagini recente » Cod sursa (job #2214059) | Cod sursa (job #148220) | Cod sursa (job #2272802) | Cod sursa (job #427817) | Cod sursa (job #339941)
Cod sursa(job #339941)
#include <stdio.h>
#include <algorithm>
using namespace std;
#define MAX_N 10010
#define MAX_K 1024
int n, k, t, l, dr, sol;
int A[MAX_N], sum[MAX_N];
int d[MAX_N], c[2][MAX_N];
void cit() {
freopen("ferma.in", "r", stdin);
freopen("ferma.out", "w", stdout);
scanf("%d %d", &n, &k);
for (int i = 1; i <= n; i++) {
scanf("%d", &A[i]);
sum[i] = sum[i - 1] + A[i];
}
}
void push(int poz) {
d[++dr] = poz;
while (c[1 - l][d[dr]] > c[1 - l][d[dr - 1]] + sum[poz] - sum[d[dr - 1]] && dr > 1) {
d[dr - 1] = d[dr];
dr--;
}
}
void solve() {
for (t = 0; t < 2; t++) {
//t = 0 nu iau primul element, t = 1 iau primul element
l = 0;
for (int j = 1; j <= k; j++) {
l = 1 - l; dr = 0;
for (int i = j; i <= n; i++) {
if (t && j == k && i == n) break;
if (t && j == 1) d[1] = 0;
c[l][i] = max(c[1 - l][d[1]] + sum[i] - sum[d[1]], c[l][i - 1]);
push(i);
}
if (t && j == k)
for (int i = k; i < n; i++)
c[l][n] = max(c[l][n], c[l][i] + sum[n] - sum[i]);
if (j == k)
sol = max(sol, c[l][n]);
else
for (int i = 1; i <= n; i++)
c[1 - l][i] = d[i] = 0;
}
for (int i = 1; i <= n; i++)
c[l][i] = 0;
}
printf("%d\n", sol);
}
int main() {
cit();
solve();
return 0;
}