Cod sursa(job #339944)

Utilizator savimSerban Andrei Stan savim Data 12 august 2009 13:12:12
Problema Ferma Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.21 kb
#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[MAX_K][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[l - 1][d[dr]] > c[l - 1][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 = j; 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[l - 1][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]);
		}                        
	}

	printf("%d\n", sol);
}

int main() {

	cit();
	solve();
	
	return 0;
}