Pagini recente » Cod sursa (job #331784) | Cod sursa (job #1494894)
#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!