#include "message.h"
#include "kanapka.h"
#include <algorithm>
#include <iostream>
using namespace std;
using LL = long long;
struct Data
{
LL total;
LL best;
LL prefix;
LL sufix;
void SendData(int node) { PutLL(node, total); PutLL(node, best); PutLL(node, prefix); PutLL(node, sufix); Send(node); };
static Data ReceiveData(int node) { Receive(node); Data res; res.total = GetLL(node); res.best = GetLL(node); res.prefix = GetLL(node); res.sufix = GetLL(node); return res; };
};
inline Data operator+(const Data &a, const Data &b)
{
Data res;
res.total = a.total + b.total;
res.best = min(a.best, b.best);
res.best = min(res.best, a.sufix + b.prefix);
res.prefix = min(a.prefix, a.total + b.prefix);
res.sufix = min(b.sufix, b.total + a.sufix);
return res;
}
inline Data Compute(LL i)
{
LL taste = GetTaste(i);
Data res;
res.total = taste;
res.best = min(taste, 0LL);
res.prefix = res.best;
res.sufix = res.best;
return res;
}
Data Compute(LL start, LL end)
{
Data res = Compute(start);
for (LL i = start + 1; i < end; ++i)
{
res = res + Compute(i);
}
return res;
}
int main()
{
int NNODES = NumberOfNodes();
int NODE = MyNodeId();
LL N = GetN();
if (NNODES > N)
NNODES = (int)N;
if (NODE >= NNODES)
return 0;
LL start = N * NODE / NNODES;
LL end = N * (NODE + 1) / NNODES;
Data data = Compute(start, end);
if (NODE != 0)
{
data.SendData(0);
}
else
{
for (int i = 1; i < NNODES; ++i)
{
data = data + Data::ReceiveData(i);
}
LL res = data.total - data.best;
cout << res;
}
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 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | #include "message.h" #include "kanapka.h" #include <algorithm> #include <iostream> using namespace std; using LL = long long; struct Data { LL total; LL best; LL prefix; LL sufix; void SendData(int node) { PutLL(node, total); PutLL(node, best); PutLL(node, prefix); PutLL(node, sufix); Send(node); }; static Data ReceiveData(int node) { Receive(node); Data res; res.total = GetLL(node); res.best = GetLL(node); res.prefix = GetLL(node); res.sufix = GetLL(node); return res; }; }; inline Data operator+(const Data &a, const Data &b) { Data res; res.total = a.total + b.total; res.best = min(a.best, b.best); res.best = min(res.best, a.sufix + b.prefix); res.prefix = min(a.prefix, a.total + b.prefix); res.sufix = min(b.sufix, b.total + a.sufix); return res; } inline Data Compute(LL i) { LL taste = GetTaste(i); Data res; res.total = taste; res.best = min(taste, 0LL); res.prefix = res.best; res.sufix = res.best; return res; } Data Compute(LL start, LL end) { Data res = Compute(start); for (LL i = start + 1; i < end; ++i) { res = res + Compute(i); } return res; } int main() { int NNODES = NumberOfNodes(); int NODE = MyNodeId(); LL N = GetN(); if (NNODES > N) NNODES = (int)N; if (NODE >= NNODES) return 0; LL start = N * NODE / NNODES; LL end = N * (NODE + 1) / NNODES; Data data = Compute(start, end); if (NODE != 0) { data.SendData(0); } else { for (int i = 1; i < NNODES; ++i) { data = data + Data::ReceiveData(i); } LL res = data.total - data.best; cout << res; } return 0; } |
English