#include "teatr.h" #include "message.h" #include <bits/stdc++.h> #define ll long long using namespace std; //int GetN(){return 2e6;} //int GetElement(int k){return (k % 5) + 1;} struct xd{ int akt; int tab[1<<21]; int base=1<<20; int query(int aa, int bb) { int akt = base + bb; int wynik = 0; while (akt > 1) { if (akt % 2 == 0) wynik += tab[akt + 1]; akt /= 2; } akt = base + aa; while (akt > 1) { if (akt % 2 == 1) wynik += tab[akt - 1]; akt /= 2; } return tab[1] - wynik; } void add(int aa,int bb) { int akt = base + aa; while(akt>=1) { tab[akt]+=bb; akt/=2; } } }; xd tree; int cnt[1<<21], t[1<<21], mx, dodaj[1<<21]; ll res; int main() { int number = NumberOfNodes(); int id = MyNodeId(); int n = GetN(); if(n < 2e6) { if(id > 0) return 0; for(int i = 0; i < n; i++) { int k = GetElement(i); res += tree.query(k + 1, 1e6 + 1); tree.add(k, 1); } cout << res; return 0; } int ile = (n - 1) / number + 1; int l = id * ile; int p = min(l + ile, n); for(int i = l; i < p; i++) { t[i - l] = GetElement(i); cnt[t[i - l]]++; } for(int i = id + 1; i < number; i++) { for(int j = 1; j <= 5; j++) { PutInt(i, cnt[j]); Send(i); } } for(int i = 0; i < id; i++) { for(int j = 1; j <= 5; j++) { Receive(i); dodaj[j] += GetInt(i); } } for(int j = 1; j <= 5; j++) tree.add(j, dodaj[j]); for(int i = l; i < p; i++) { res += tree.query(t[i - l] + 1, 1e6 + 1); tree.add(t[i - l], 1); } if(id != 0) { PutLL(0, res); Send(0); return 0; } if(id == 0) { for(int i = 1; i < number; i++) { Receive(i); res += GetLL(i); } cout << res; 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 | #include "teatr.h" #include "message.h" #include <bits/stdc++.h> #define ll long long using namespace std; //int GetN(){return 2e6;} //int GetElement(int k){return (k % 5) + 1;} struct xd{ int akt; int tab[1<<21]; int base=1<<20; int query(int aa, int bb) { int akt = base + bb; int wynik = 0; while (akt > 1) { if (akt % 2 == 0) wynik += tab[akt + 1]; akt /= 2; } akt = base + aa; while (akt > 1) { if (akt % 2 == 1) wynik += tab[akt - 1]; akt /= 2; } return tab[1] - wynik; } void add(int aa,int bb) { int akt = base + aa; while(akt>=1) { tab[akt]+=bb; akt/=2; } } }; xd tree; int cnt[1<<21], t[1<<21], mx, dodaj[1<<21]; ll res; int main() { int number = NumberOfNodes(); int id = MyNodeId(); int n = GetN(); if(n < 2e6) { if(id > 0) return 0; for(int i = 0; i < n; i++) { int k = GetElement(i); res += tree.query(k + 1, 1e6 + 1); tree.add(k, 1); } cout << res; return 0; } int ile = (n - 1) / number + 1; int l = id * ile; int p = min(l + ile, n); for(int i = l; i < p; i++) { t[i - l] = GetElement(i); cnt[t[i - l]]++; } for(int i = id + 1; i < number; i++) { for(int j = 1; j <= 5; j++) { PutInt(i, cnt[j]); Send(i); } } for(int i = 0; i < id; i++) { for(int j = 1; j <= 5; j++) { Receive(i); dodaj[j] += GetInt(i); } } for(int j = 1; j <= 5; j++) tree.add(j, dodaj[j]); for(int i = l; i < p; i++) { res += tree.query(t[i - l] + 1, 1e6 + 1); tree.add(t[i - l], 1); } if(id != 0) { PutLL(0, res); Send(0); return 0; } if(id == 0) { for(int i = 1; i < number; i++) { Receive(i); res += GetLL(i); } cout << res; return 0; } } |