Cod sursa(job #1494894)

Utilizator iordache.bogdanIordache Ioan-Bogdan iordache.bogdan Data 1 octombrie 2015 22:54:49
Problema Ferma Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.71 kb
#include <fstream>
#include <algorithm>
#include <cstring>
#include <vector>

#define DIM 20005
#define infile "ferma.in"
#define outfile "ferma.out"

std::ifstream fin(infile);
std::ofstream fout(outfile);

const int inf = 2000000000;

int chickensProductivity[DIM];

int partialSum[DIM];

int deque[DIM];

int main() {

	int chickenCount, collectCount;

	fin >> chickenCount >> collectCount;

	for (int chicken = 1; chicken <= chickenCount; ++chicken) {

		fin >> chickensProductivity[chicken];

		chickensProductivity[chicken + chickenCount] = chickensProductivity[chicken];

	}


	int solution = 0;

	for (int collect = 1; collect <= collectCount; ++collect) {

		int left = 1, right = 1;

		deque[1] = 0;

		int currAns = -inf, leftEndAns, rightEndAns;

		for (int chicken = 1; chicken < 2 * chickenCount; ++chicken) {

			partialSum[chicken] = partialSum[chicken - 1] + chickensProductivity[chicken];

			if (left <= right && chicken - deque[left] == chickenCount) {

				++left;

			}


			if (partialSum[chicken] - partialSum[deque[left]] > currAns) {

				currAns = partialSum[chicken] - partialSum[deque[left]];

				leftEndAns = deque[left] + 1;

				rightEndAns = chicken;

			}

			while (left <= right && partialSum[chicken] <= partialSum[deque[right]]) {

				--right;

			}

			deque[++right] = chicken;

		}

		for (int chicken = leftEndAns; chicken <= rightEndAns; ++chicken) {

			int temp = (chicken <= chickenCount ? chicken : chicken - chickenCount);

			chickensProductivity[temp] *= -1;
			chickensProductivity[temp + chickenCount] *= -1;

		}

		solution += currAns;

	}


	fout << solution;

	return 0;

}

//Trust me, I'm the Doctor!