Cod sursa(job #758173)

Utilizator danalex97Dan H Alexandru danalex97 Data 14 iunie 2012 19:06:19
Problema Ferma Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.04 kb
#include <cstdio>
#include <algorithm>

using namespace std;

#define Maxn 10010
#define Maxk 1030

int n,k,t,l;
int dr,sol,poz;
int A[Maxn],sum[Maxn];
int c[Maxn][2];

int main() 
{
	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];
	}

	for (t = 0; t < 2; t++)
	{
		l = 0;
		for (int j = 1; j <= k; j++)
		{
			l = 1 - l; dr = 0;
			poz = 0;
			for (int i = j; i <= n; i++) 
			{
				if (t && j == k && i == n) break;
				if (t && j == 1) poz = 0;
				c[i][l] = max(c[poz][1 - l] + sum[i] - sum[poz], c[i - 1][l]);
				if (c[i][1 - l] > c[poz][1 - l] + sum[i] - sum[poz])
					poz = i;
			}
			if (t && j == k)
				for (int i = k; i < n; i++)      	
					c[n][l] = max(c[n][l], c[i][l] + sum[n] - sum[i]);
			if (j == k)
				sol = max(sol, c[n][l]);
			for (int i = 1; i <= n; i++)
				c[i][1 - l] = 0;
		}
		for (int i = 1; i <= n; i++)
			c[i][l] = 0;
	}

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