#include "teatr.h" #include "message.h" #include <bits/stdc++.h> using namespace std; #define PB push_back #define MP make_pair #define LL long long #define int LL #define short int32_t #define FOR(i,a,b) for(int i = (a); i <= (b); i++) #define RE(i,n) FOR(i,1,n) #define REP(i,n) FOR(i,0,(int)(n)-1) #define R(i,n) REP(i,n) #define VI vector<int> #define PII pair<int,int> #define LD long double #define FI first #define SE second #define st FI #define nd SE #define ALL(x) (x).begin(), (x).end() #define SZ(x) ((int)(x).size()) void put(int a,int b){ PutInt(a,int32_t(b)); } #define PutInt dgayuwgyugugdw template<class C> void mini(C& _a4, C _b4) { _a4 = min(_a4, _b4); } template<class C> void maxi(C& _a4, C _b4) { _a4 = max(_a4, _b4); } template<class TH> void _dbg(const char *sdbg, TH h){ cerr<<sdbg<<'='<<h<<endl; } template<class TH, class... TA> void _dbg(const char *sdbg, TH h, TA... a) { while(*sdbg!=',')cerr<<*sdbg++;cerr<<'='<<h<<','; _dbg(sdbg+1, a...); } template<class T> ostream& operator<<(ostream& os, vector<T> V) { os << "["; for (auto& vv : V) os << vv << ","; os << "]"; return os; } #ifdef LOCAL #define debug(...) if(id == 0) _dbg(#__VA_ARGS__, __VA_ARGS__) #else #define debug(...) (__VA_ARGS__) #define cerr if(0)cout #endif const int MAX = 1e6 + 1; int proc,id; int dp[MAX]; int res = 0; int n; void sum(int x){ for(;x < MAX; x += (x & -x)){ res += dp[x]; } } void add(int x){ for(;x;x -= (x & -x)){ dp[x]++; } } void licz(int po, int ko){ R(i,po){ dp[GetElement(i)]++; } for(int i = MAX -1 ;i;i--){ dp[i - (i&-i)] += dp[i]; } for(int i = po; i< ko; i++){ int pom = GetElement(i); sum(pom + 1); add(pom); } PutLL(0,res); Send(0); } void lacz(){ int wyn = 0; R(i,proc){ Receive(i); wyn += GetLL(i); } cout << wyn << "\n"; } int32_t main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout << fixed << setprecision(11); cerr << fixed << setprecision(6); proc = NumberOfNodes(); id = MyNodeId(); n = GetN(); int il = (n + proc - 1) / proc; licz(min(il*id, n), min(il*(id+1),n)); if(id == 0)lacz(); }
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 | #include "teatr.h" #include "message.h" #include <bits/stdc++.h> using namespace std; #define PB push_back #define MP make_pair #define LL long long #define int LL #define short int32_t #define FOR(i,a,b) for(int i = (a); i <= (b); i++) #define RE(i,n) FOR(i,1,n) #define REP(i,n) FOR(i,0,(int)(n)-1) #define R(i,n) REP(i,n) #define VI vector<int> #define PII pair<int,int> #define LD long double #define FI first #define SE second #define st FI #define nd SE #define ALL(x) (x).begin(), (x).end() #define SZ(x) ((int)(x).size()) void put(int a,int b){ PutInt(a,int32_t(b)); } #define PutInt dgayuwgyugugdw template<class C> void mini(C& _a4, C _b4) { _a4 = min(_a4, _b4); } template<class C> void maxi(C& _a4, C _b4) { _a4 = max(_a4, _b4); } template<class TH> void _dbg(const char *sdbg, TH h){ cerr<<sdbg<<'='<<h<<endl; } template<class TH, class... TA> void _dbg(const char *sdbg, TH h, TA... a) { while(*sdbg!=',')cerr<<*sdbg++;cerr<<'='<<h<<','; _dbg(sdbg+1, a...); } template<class T> ostream& operator<<(ostream& os, vector<T> V) { os << "["; for (auto& vv : V) os << vv << ","; os << "]"; return os; } #ifdef LOCAL #define debug(...) if(id == 0) _dbg(#__VA_ARGS__, __VA_ARGS__) #else #define debug(...) (__VA_ARGS__) #define cerr if(0)cout #endif const int MAX = 1e6 + 1; int proc,id; int dp[MAX]; int res = 0; int n; void sum(int x){ for(;x < MAX; x += (x & -x)){ res += dp[x]; } } void add(int x){ for(;x;x -= (x & -x)){ dp[x]++; } } void licz(int po, int ko){ R(i,po){ dp[GetElement(i)]++; } for(int i = MAX -1 ;i;i--){ dp[i - (i&-i)] += dp[i]; } for(int i = po; i< ko; i++){ int pom = GetElement(i); sum(pom + 1); add(pom); } PutLL(0,res); Send(0); } void lacz(){ int wyn = 0; R(i,proc){ Receive(i); wyn += GetLL(i); } cout << wyn << "\n"; } int32_t main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout << fixed << setprecision(11); cerr << fixed << setprecision(6); proc = NumberOfNodes(); id = MyNodeId(); n = GetN(); int il = (n + proc - 1) / proc; licz(min(il*id, n), min(il*(id+1),n)); if(id == 0)lacz(); } |