#include <cstdio> #include <cstring> #include <cmath> #include <cassert> #include <iostream> #include <algorithm> #include <iterator> #include <string> #include <vector> #include <queue> #include <bitset> #include <utility> #include <stack> using namespace std; typedef long long LL; typedef pair<int,int> PII; typedef vector<int> VI; #define MP make_pair #define FOR(v,p,k) for(int v=(p);v<=(k);++v) #define FORD(v,p,k) for(int v=(p);v>=(k);--v) #define REP(i,n) for(int i=0;i<(n);++i) #define VAR(v,i) __typeof(i) v=(i) #define FOREACH(i,c) for(VAR(i,(c).begin());i!=(c).end();++i) #define PB push_back #define ST first #define ND second #define SIZE(x) (int)x.size() #define ALL(c) c.begin(),c.end() #define ODD(x) ((x)%2) #define EVEN(x) (!(ODD(x))) #include "maklib.h" #include "message.h" class SegmentParams { LL maxPrefixSum, maxSubSum, maxSuffixSum, sum; public: SegmentParams() : maxPrefixSum(0), maxSubSum(0), maxSuffixSum(0), sum(0) {} void printMaxSubSum() { printf("%lld\n", maxSubSum); } void extendRight(LL value) { sum = sum + value; maxPrefixSum = max(maxPrefixSum, sum); maxSuffixSum = max(0LL, maxSuffixSum + value); maxSubSum = max(maxSubSum, maxSuffixSum); } void extendRight(const SegmentParams &other) { maxSubSum = max(max(maxSubSum, other.maxSubSum), maxSuffixSum + other.maxPrefixSum); maxPrefixSum = max(maxPrefixSum, sum + other.maxPrefixSum); maxSuffixSum = max(maxSuffixSum + other.sum, other.maxSuffixSum); sum = sum + other.sum; } void send() { PutLL(0, maxPrefixSum); PutLL(0, maxSubSum); PutLL(0, maxSuffixSum); PutLL(0, sum); Send(0); } void fillFromNode(int nodeID) { maxPrefixSum = GetLL(nodeID); maxSubSum = GetLL(nodeID); maxSuffixSum = GetLL(nodeID); sum = GetLL(nodeID); } }; int main() { int nodesNumber = NumberOfNodes(); int nodeID = MyNodeId(); int n = Size(); int elementsPerNode = n/nodesNumber; int firstElementPos = 1 + nodeID*elementsPerNode; int lastElementPos = firstElementPos + elementsPerNode - 1; if (nodeID == nodesNumber-1) { lastElementPos = n; } VI nodeElements; int nodeElementsSize = lastElementPos - firstElementPos + 1; nodeElements.reserve(nodeElementsSize); FOR(pos, firstElementPos, lastElementPos) { nodeElements.PB(ElementAt(pos)); } SegmentParams nodeSegmentParams; REP(i, nodeElementsSize) { nodeSegmentParams.extendRight(nodeElements[i]); } if (nodeID > 0) { nodeSegmentParams.send(); } else { vector<SegmentParams> allSegmentParams(nodesNumber); allSegmentParams[0] = nodeSegmentParams; REP(i, nodesNumber-1) { int receivedNodeID = Receive(-1); allSegmentParams[receivedNodeID].fillFromNode(receivedNodeID); } SegmentParams accumulatedSegmentParams = nodeSegmentParams; FOR(otherNodeID, 1, nodesNumber-1) { accumulatedSegmentParams.extendRight(allSegmentParams[otherNodeID]); } accumulatedSegmentParams.printMaxSubSum(); } 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 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 | #include <cstdio> #include <cstring> #include <cmath> #include <cassert> #include <iostream> #include <algorithm> #include <iterator> #include <string> #include <vector> #include <queue> #include <bitset> #include <utility> #include <stack> using namespace std; typedef long long LL; typedef pair<int,int> PII; typedef vector<int> VI; #define MP make_pair #define FOR(v,p,k) for(int v=(p);v<=(k);++v) #define FORD(v,p,k) for(int v=(p);v>=(k);--v) #define REP(i,n) for(int i=0;i<(n);++i) #define VAR(v,i) __typeof(i) v=(i) #define FOREACH(i,c) for(VAR(i,(c).begin());i!=(c).end();++i) #define PB push_back #define ST first #define ND second #define SIZE(x) (int)x.size() #define ALL(c) c.begin(),c.end() #define ODD(x) ((x)%2) #define EVEN(x) (!(ODD(x))) #include "maklib.h" #include "message.h" class SegmentParams { LL maxPrefixSum, maxSubSum, maxSuffixSum, sum; public: SegmentParams() : maxPrefixSum(0), maxSubSum(0), maxSuffixSum(0), sum(0) {} void printMaxSubSum() { printf("%lld\n", maxSubSum); } void extendRight(LL value) { sum = sum + value; maxPrefixSum = max(maxPrefixSum, sum); maxSuffixSum = max(0LL, maxSuffixSum + value); maxSubSum = max(maxSubSum, maxSuffixSum); } void extendRight(const SegmentParams &other) { maxSubSum = max(max(maxSubSum, other.maxSubSum), maxSuffixSum + other.maxPrefixSum); maxPrefixSum = max(maxPrefixSum, sum + other.maxPrefixSum); maxSuffixSum = max(maxSuffixSum + other.sum, other.maxSuffixSum); sum = sum + other.sum; } void send() { PutLL(0, maxPrefixSum); PutLL(0, maxSubSum); PutLL(0, maxSuffixSum); PutLL(0, sum); Send(0); } void fillFromNode(int nodeID) { maxPrefixSum = GetLL(nodeID); maxSubSum = GetLL(nodeID); maxSuffixSum = GetLL(nodeID); sum = GetLL(nodeID); } }; int main() { int nodesNumber = NumberOfNodes(); int nodeID = MyNodeId(); int n = Size(); int elementsPerNode = n/nodesNumber; int firstElementPos = 1 + nodeID*elementsPerNode; int lastElementPos = firstElementPos + elementsPerNode - 1; if (nodeID == nodesNumber-1) { lastElementPos = n; } VI nodeElements; int nodeElementsSize = lastElementPos - firstElementPos + 1; nodeElements.reserve(nodeElementsSize); FOR(pos, firstElementPos, lastElementPos) { nodeElements.PB(ElementAt(pos)); } SegmentParams nodeSegmentParams; REP(i, nodeElementsSize) { nodeSegmentParams.extendRight(nodeElements[i]); } if (nodeID > 0) { nodeSegmentParams.send(); } else { vector<SegmentParams> allSegmentParams(nodesNumber); allSegmentParams[0] = nodeSegmentParams; REP(i, nodesNumber-1) { int receivedNodeID = Receive(-1); allSegmentParams[receivedNodeID].fillFromNode(receivedNodeID); } SegmentParams accumulatedSegmentParams = nodeSegmentParams; FOR(otherNodeID, 1, nodesNumber-1) { accumulatedSegmentParams.extendRight(allSegmentParams[otherNodeID]); } accumulatedSegmentParams.printMaxSubSum(); } return 0; } |