#include "message.h" #include "kanapka.h" #include <algorithm> #include <iostream> using namespace std; int NUM_NODES; int NODE; template<class T> void Put(int target, const T &value) { for (size_t i = 0; i < sizeof(T); ++i) { PutChar(target, reinterpret_cast<const char*>(&value)[i]); } } template<class T> void Send(int target, const T &value) { Put(target, value); Send(target); } template<class T> void Get(int source, T &value) { for (size_t i = 0; i < sizeof(T); ++i) { reinterpret_cast<char*>(&value)[i] = GetChar(source); } } template<class T> int Receive(int source, T &value) { source = Receive(source); Get(source, value); return source; } // --------------- struct Data { long long sum; long long left; long long right; long long both; }; inline Data Singleton(long long x) { Data data; data.sum = x; data.left = max(x, 0LL); data.right = data.left; data.both = data.left; return data; } Data Combine(const Data &a, const Data &b) { Data data; data.sum = a.sum + b.sum; data.left = max(a.left, a.sum + b.left); data.right = max(b.right, b.sum + a.right); data.both = max(max(a.both + b.sum, b.both + a.sum), a.left + b.right); return data; } Data Interval(long long a, long long b) { Data data = Singleton(GetTaste(a)); for (long long i = a+1; i < b; ++i) { data = Combine(data, Singleton(GetTaste(i))); } return data; } int main() { int NUM_NODES = NumberOfNodes(); int NODE = MyNodeId(); long long N = GetN(); if (NUM_NODES > N) NUM_NODES = N; if (NODE >= NUM_NODES) return 0; Data data = Interval(NODE * N / NUM_NODES, (NODE+1) * N / NUM_NODES); if (NODE > 0) { Send(0, data); } else { for (int i=1;i<NUM_NODES;++i) { Data data2; Receive(i, data2); data = Combine(data, data2); } cout << data.both << "\n"; } }
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 | #include "message.h" #include "kanapka.h" #include <algorithm> #include <iostream> using namespace std; int NUM_NODES; int NODE; template<class T> void Put(int target, const T &value) { for (size_t i = 0; i < sizeof(T); ++i) { PutChar(target, reinterpret_cast<const char*>(&value)[i]); } } template<class T> void Send(int target, const T &value) { Put(target, value); Send(target); } template<class T> void Get(int source, T &value) { for (size_t i = 0; i < sizeof(T); ++i) { reinterpret_cast<char*>(&value)[i] = GetChar(source); } } template<class T> int Receive(int source, T &value) { source = Receive(source); Get(source, value); return source; } // --------------- struct Data { long long sum; long long left; long long right; long long both; }; inline Data Singleton(long long x) { Data data; data.sum = x; data.left = max(x, 0LL); data.right = data.left; data.both = data.left; return data; } Data Combine(const Data &a, const Data &b) { Data data; data.sum = a.sum + b.sum; data.left = max(a.left, a.sum + b.left); data.right = max(b.right, b.sum + a.right); data.both = max(max(a.both + b.sum, b.both + a.sum), a.left + b.right); return data; } Data Interval(long long a, long long b) { Data data = Singleton(GetTaste(a)); for (long long i = a+1; i < b; ++i) { data = Combine(data, Singleton(GetTaste(i))); } return data; } int main() { int NUM_NODES = NumberOfNodes(); int NODE = MyNodeId(); long long N = GetN(); if (NUM_NODES > N) NUM_NODES = N; if (NODE >= NUM_NODES) return 0; Data data = Interval(NODE * N / NUM_NODES, (NODE+1) * N / NUM_NODES); if (NODE > 0) { Send(0, data); } else { for (int i=1;i<NUM_NODES;++i) { Data data2; Receive(i, data2); data = Combine(data, data2); } cout << data.both << "\n"; } } |