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
//Autor: Mateusz Wasylkiewicz
//Zawody: Potyczki Algorytmiczne 2018
//Strona: http://potyczki.mimuw.edu.pl/
//Zadanie: Teatr, runda 4B
//Czas: O(N/NumberOfNodes()*M+(NumberOfNodes()*M)^2)
#include <bits/stdc++.h>
#include "teatr.h"
#include "message.h"
using namespace std;

typedef long long LL;
#define FOR(x, a, b) for (LL x = (a); x <= (b); x++)
#define REP(x, n) for (LL x = 0; x < (n); x++)

const LL KOMPUTERY = 100, M = 5; //GetElement(k) <= M
LL N, id, pocz, kon, wynik = 0, licz[KOMPUTERY][M];

void wczytaj_dane_poczatkowe()
{
	id = LL(MyNodeId());
	N = LL(GetN());
	pocz = (id * N) / KOMPUTERY;
	kon = ((id + 1) * N) / KOMPUTERY - 1;
}

void rozwiaz_przedzial()
{
	FOR(i, pocz, kon)
	{
		LL el = LL(GetElement(i)) - 1;
		FOR(j, el + 1, M - 1)
			wynik += licz[0][j];
		licz[0][el]++;
	}
}

void wyslij_dane()
{
	PutLL(0, wynik);
	REP(j, M)
		PutLL(0, licz[0][j]);
	Send(0);
}

void odbierz_dane()
{
	FOR(t, 1, KOMPUTERY - 1)
	{
		Receive(t);
		wynik += GetLL(t);
		REP(j, M)
			licz[t][j] = GetLL(t);
	}
}

void rozwiaz_pomiedzy()
{
	REP(t2, KOMPUTERY)
		REP(t1, t2)
			REP(j1, M)
				REP(j2, j1)
					wynik += licz[t1][j1] * licz[t2][j2];
}

void zrob_test()
{
	wczytaj_dane_poczatkowe();
	rozwiaz_przedzial();
	if (id == 0)
	{
		odbierz_dane();
		rozwiaz_pomiedzy();
		cout << wynik << '\n';
	}
	else
		wyslij_dane();
}

int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	zrob_test();
	return 0;
}