#include "message.h"
#include "teatr.h"
#include "bits/stdc++.h"
using namespace std;
#define VMAX 1000000LL
#define POT 262144
int drzewo[POT*2] = { 0 };
int lisci = 1;
void aktualizuj(int e){
int i = lisci + e;
while(i>0){
drzewo[i] ++;
i /= 2;
}
}
int iwiek(int e){
int i = lisci + e;
int wyn = 0;
while(i > 1){
if(i%2 == 0)
wyn += drzewo[i+1];
i/=2;
}
return wyn;
}
int main() {
int N=GetN(),Nd=NumberOfNodes(),Id=MyNodeId(), e;
int m = (VMAX*Id)/Nd, g = (VMAX*(Id+1))/Nd;
int ile_wiekszych = 0;
long long wyn = 0;
while(lisci<g-m)
lisci *= 2;
for(int i=0; i<N; i++){
e = GetElement(i);
if(e>=g)
ile_wiekszych++;
else if(e>=m){
aktualizuj(e-m);
wyn += iwiek(e-m) + ile_wiekszych;
}
}
//fprintf(stderr,"Id:%d m=%d g=%d wyn=%lld\n",Id,m,g,wyn);
if(Id > 0){
PutLL(0,wyn);
Send(0);
}else{
for(int i=1;i<Nd;i++){
Receive(i);
wyn += GetLL(i);
}
printf("%lld",wyn);
}
}
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 | #include "message.h" #include "teatr.h" #include "bits/stdc++.h" using namespace std; #define VMAX 1000000LL #define POT 262144 int drzewo[POT*2] = { 0 }; int lisci = 1; void aktualizuj(int e){ int i = lisci + e; while(i>0){ drzewo[i] ++; i /= 2; } } int iwiek(int e){ int i = lisci + e; int wyn = 0; while(i > 1){ if(i%2 == 0) wyn += drzewo[i+1]; i/=2; } return wyn; } int main() { int N=GetN(),Nd=NumberOfNodes(),Id=MyNodeId(), e; int m = (VMAX*Id)/Nd, g = (VMAX*(Id+1))/Nd; int ile_wiekszych = 0; long long wyn = 0; while(lisci<g-m) lisci *= 2; for(int i=0; i<N; i++){ e = GetElement(i); if(e>=g) ile_wiekszych++; else if(e>=m){ aktualizuj(e-m); wyn += iwiek(e-m) + ile_wiekszych; } } //fprintf(stderr,"Id:%d m=%d g=%d wyn=%lld\n",Id,m,g,wyn); if(Id > 0){ PutLL(0,wyn); Send(0); }else{ for(int i=1;i<Nd;i++){ Receive(i); wyn += GetLL(i); } printf("%lld",wyn); } } |
English