#include <iostream>
#include <vector>
#include "message.h"
#include "kanapka.h"
struct sub_res {
int best;
int left;
int right;
};
const sub_res max_subarray(const std::vector<long long int> &ar) {
int left = 0;
int current_left = 0;
int right = 0;
int max = 0;
int acc = 0;
for (long long el : ar) {
if (el > 0 && acc == 0)
current_left = left;
left += el;
acc += el;
if (acc > max) {
max = acc;
right = 0;
} else
right += el;
if (acc < 0)
acc = 0;
}
return {max, current_left, right};
}
int main() {
int N = NumberOfNodes();
int node = MyNodeId();
long long int n = GetN();
long long lenPerNode = n / N;
long long start = lenPerNode * node;
long long end = (node == N - 1) ? n : lenPerNode * (node + 1);
long long int size = end - start;
std::vector<long long int> data((unsigned long) size);
for (int i = 0; i < size; i++)
data[i] = -GetTaste(start + i);
const sub_res &res = max_subarray(data);
if (node != 0) {
PutLL(0, res.left);
PutLL(0, res.best);
PutLL(0, res.right);
Send(0);
} else {
std::vector<long long int> all((unsigned long long)(N * 3));
all[0] = res.left;
all[1] = res.best;
all[2] = res.right;
for (int i = 1; i < N; i++) {
const int src = Receive(-1);
all[src * 3] = GetLL(src);
all[src * 3 + 1] = GetLL(src);
all[src * 3 + 2] = GetLL(src);
}
const sub_res &all_res = max_subarray(all);
std::cout << (-all_res.left + -all_res.right) << std::endl;
}
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 | #include <iostream> #include <vector> #include "message.h" #include "kanapka.h" struct sub_res { int best; int left; int right; }; const sub_res max_subarray(const std::vector<long long int> &ar) { int left = 0; int current_left = 0; int right = 0; int max = 0; int acc = 0; for (long long el : ar) { if (el > 0 && acc == 0) current_left = left; left += el; acc += el; if (acc > max) { max = acc; right = 0; } else right += el; if (acc < 0) acc = 0; } return {max, current_left, right}; } int main() { int N = NumberOfNodes(); int node = MyNodeId(); long long int n = GetN(); long long lenPerNode = n / N; long long start = lenPerNode * node; long long end = (node == N - 1) ? n : lenPerNode * (node + 1); long long int size = end - start; std::vector<long long int> data((unsigned long) size); for (int i = 0; i < size; i++) data[i] = -GetTaste(start + i); const sub_res &res = max_subarray(data); if (node != 0) { PutLL(0, res.left); PutLL(0, res.best); PutLL(0, res.right); Send(0); } else { std::vector<long long int> all((unsigned long long)(N * 3)); all[0] = res.left; all[1] = res.best; all[2] = res.right; for (int i = 1; i < N; i++) { const int src = Receive(-1); all[src * 3] = GetLL(src); all[src * 3 + 1] = GetLL(src); all[src * 3 + 2] = GetLL(src); } const sub_res &all_res = max_subarray(all); std::cout << (-all_res.left + -all_res.right) << std::endl; } return 0; } |
English