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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <fstream>
#include <sstream>
#include <set>
#include <map>
#include <vector>
#include <list>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <string>
#include <queue>
#include <bitset>		//UWAGA - w czasie kompilacji musi byc znany rozmiar wektora - nie mozna go zmienic
#include <cassert>
#include <iomanip>		//do setprecision
#include <ctime>
#include <complex>
using namespace std;

#include "message.h"

#define FOR(i,b,e) for(int i=(b);i<(e);++i)
#define FORQ(i,b,e) for(int i=(b);i<=(e);++i)
#define FORD(i,b,e) for(int i=(b)-1;i>=(e);--i)
#define REP(x, n) for(int x = 0; x < (n); ++x)

#define ST first
#define ND second
#define PB push_back
#define MP make_pair
#define LL long long
#define ULL unsigned LL
#define LD long double

const double pi = 3.141592653589793238462643383279502884197169399375105820974944592307816406286208998628034825342;

// change the include here!!!
#include "teatr.h"

const int MR = 1e6 + 10; // limit for array size

int cnt[MR], v[MR + 1000];

int n;  // array size
int t[2 * MR];

void build() {  // build the tree
	for (int i = n - 1; i > 0; --i) t[i] = t[i << 1] + t[i << 1 | 1];
}

void modify(int p, int value) {  // set value at position p
	for (t[p += n] = value; p > 1; p >>= 1) t[p >> 1] = t[p] + t[p ^ 1];
}

int query(int l, int r) {  // sum on interval [l, r)
	int res = 0;
	for (l += n, r += n; l < r; l >>= 1, r >>= 1) {
		if (l & 1) res += t[l++];
		if (r & 1) res += t[--r];
	}
	return res;
}

int main()
{
	int id = MyNodeId();
	int ileI = NumberOfNodes();

	// Get number of elements
	int N = GetN();// choose main funtion which we distribute

	if (N < ileI)
		ileI = N;

	if (id >= ileI)
		return 0;

	// wylicz ile elementow bede przetwarzal
	int ile = N / ileI;
	int beg = id * ile;

	if (id == ileI - 1)
	{
		ile += N % ileI;
	}
	
	//clock_t tm = clock();

	// wez wartosci sprzed swojego przedzialu - w najgorszym razie to zajmie 3,5 s
	REP(i, beg)
		cnt[GetElement(i)]++;

	/*tm = clock() - tm;
	printf("It took me %d clicks (%lf seconds).\n", tm, ((double)tm) / CLOCKS_PER_SEC);*/

	// wylicz swoje wartosci
	REP(i, ile)
		v[i] = GetElement(beg + i);

	//printf("moje wartosci: (%d, %d) (%d, %d) (%d, %d) (%d, %d) (%d, %d)\n", precnt[1], cnt[1], precnt[2], cnt[2], precnt[3], cnt[3], precnt[4], cnt[4], precnt[5], cnt[5]);

	// rozmiar tablicy - wszystkich wartosci moze byc max 10^6
	n = MR - 10;

	// wrzuc do drzewa czestosci poprzednich wystapien
	REP(i, n)
		t[n + i] = cnt[i + 1];
	build();

	LL res = 0;

	REP(i, ile)
	{
		cnt[v[i]]++;
		modify(v[i] - 1, cnt[v[i]]);

		// policz ile wiekszych liczb przed nia sie znalazlo
		res += query(v[i], n);
	}

	// odbierz wynik od poprzedniego
	if (id)
	{
		Receive(id - 1);
		res += GetLL(id - 1);
	}

	if (id == ileI - 1)
	{
		// output - przez ostatni komputer
		printf("%lld\n", res);
	}
	else
	{
		// przeslij dalej
		PutLL(id + 1, res);
		Send(id + 1);
	}

	return 0;
}