#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; } |