#include"maklib.h"
#include"message.h"
#include<iostream>
#include<algorithm>
#include<vector>
#include<stdint.h>
using namespace std;
vector<int64_t> sums;
int64_t maxv=0;
void addResult(int64_t v) {
if(v>maxv) maxv=v;
sums.push_back(v);
while(sums.size()>2) {
int64_t a=sums[sums.size()-3];
int64_t b=sums[sums.size()-2];
int64_t c=sums[sums.size()-1];
if((a>=-b)&&(c>=-b)) {
v=a+b+c;
sums.pop_back(); sums.pop_back(); sums.pop_back();
sums.push_back(v);
if(v>maxv) maxv=v;
} else break; // nie ma optymalizacji przerywamy
}
}
/** Przetwarza podciąg tablicy */
void process(int from, int to) {
int64_t sum=0;
for(int i=from;i<to;++i) {
int v=ElementAt(i+1);
//cerr<<"Sum="<<sum<<", E["<<(i+1)<<"]="<<v<<endl;
if(sum==0) sum=v;
else if(sum>0) {
if(v>=0) sum+=v;
else {
addResult(sum); sum=v;
}
} else {
if(v<=0) sum+=v;
else {
addResult(sum); sum=v;
}
}
}
if(sum>0) addResult(sum);
}
int main() {
if((Size()<1000)||(NumberOfNodes()==1)) { // dla malej ilosci danych liczymy na jednym nodzie
if(MyNodeId()!=0) return 0;
cerr<<"Single node mode"<<endl;
process(0,Size());
cout<<maxv<<endl;
return 0;
}
// dla wiekszej ilosci danych/nodow
int from=(MyNodeId()*Size())/NumberOfNodes();
int to=((MyNodeId()+1)*Size())/NumberOfNodes();
// kazy z nodow przetwarza swoje dane
process(from,to);
// TODO: Jeszcze bardziej rozproszenie
// i przekazuje wyniki do glownego
if(MyNodeId()!=0) {
// oczekujemy na polecenie wysylania wyniku, aby nie zanieczyszczac buforow
Receive(0);
PutLL(0,maxv); // wysylamym maximum
PutInt(0,sums.size());
for(int i=0;i<sums.size();++i) PutLL(0,sums[i]);
Send(0);
return 0;
}
// zbieranie wynikow od pozostalych nodow
for(int i=1;i<NumberOfNodes();++i) {
Send(i); // dajemy komunikat, ze bedziemy odbierac dane
Receive(i);
int64_t v=GetLL(i);
if(v>maxv) maxv=v;
int size=GetInt(i);
for(int j=0;j<size;++j) {
v=GetLL(i);
addResult(v);
}
}
cout<<maxv<<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 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | #include"maklib.h" #include"message.h" #include<iostream> #include<algorithm> #include<vector> #include<stdint.h> using namespace std; vector<int64_t> sums; int64_t maxv=0; void addResult(int64_t v) { if(v>maxv) maxv=v; sums.push_back(v); while(sums.size()>2) { int64_t a=sums[sums.size()-3]; int64_t b=sums[sums.size()-2]; int64_t c=sums[sums.size()-1]; if((a>=-b)&&(c>=-b)) { v=a+b+c; sums.pop_back(); sums.pop_back(); sums.pop_back(); sums.push_back(v); if(v>maxv) maxv=v; } else break; // nie ma optymalizacji przerywamy } } /** Przetwarza podciąg tablicy */ void process(int from, int to) { int64_t sum=0; for(int i=from;i<to;++i) { int v=ElementAt(i+1); //cerr<<"Sum="<<sum<<", E["<<(i+1)<<"]="<<v<<endl; if(sum==0) sum=v; else if(sum>0) { if(v>=0) sum+=v; else { addResult(sum); sum=v; } } else { if(v<=0) sum+=v; else { addResult(sum); sum=v; } } } if(sum>0) addResult(sum); } int main() { if((Size()<1000)||(NumberOfNodes()==1)) { // dla malej ilosci danych liczymy na jednym nodzie if(MyNodeId()!=0) return 0; cerr<<"Single node mode"<<endl; process(0,Size()); cout<<maxv<<endl; return 0; } // dla wiekszej ilosci danych/nodow int from=(MyNodeId()*Size())/NumberOfNodes(); int to=((MyNodeId()+1)*Size())/NumberOfNodes(); // kazy z nodow przetwarza swoje dane process(from,to); // TODO: Jeszcze bardziej rozproszenie // i przekazuje wyniki do glownego if(MyNodeId()!=0) { // oczekujemy na polecenie wysylania wyniku, aby nie zanieczyszczac buforow Receive(0); PutLL(0,maxv); // wysylamym maximum PutInt(0,sums.size()); for(int i=0;i<sums.size();++i) PutLL(0,sums[i]); Send(0); return 0; } // zbieranie wynikow od pozostalych nodow for(int i=1;i<NumberOfNodes();++i) { Send(i); // dajemy komunikat, ze bedziemy odbierac dane Receive(i); int64_t v=GetLL(i); if(v>maxv) maxv=v; int size=GetInt(i); for(int j=0;j<size;++j) { v=GetLL(i); addResult(v); } } cout<<maxv<<endl; return 0; } |
English