CSES - Datatähti 2021 alku - Results
Submission details
Task:Arpakuutiot
Sender:hltk
Submission time:2020-09-28 02:18:24 +0300
Language:C++17
Status:READY
Result:35
Feedback
groupverdictscore
#1ACCEPTED35
#20
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2details
#2ACCEPTED0.01 s1, 2details
#3ACCEPTED0.01 s1, 2details
#4ACCEPTED0.01 s1, 2details
#5ACCEPTED0.01 s1, 2details
#6ACCEPTED0.01 s1, 2details
#7ACCEPTED0.01 s1, 2details
#8ACCEPTED0.01 s1, 2details
#9ACCEPTED0.01 s1, 2details
#10ACCEPTED0.01 s1, 2details
#110.01 s2details
#120.01 s2details
#13--2details
#14ACCEPTED0.01 s2details
#150.01 s2details
#16ACCEPTED0.01 s2details
#17ACCEPTED0.01 s2details
#18--2details
#19ACCEPTED0.01 s2details
#20ACCEPTED0.01 s2details
#210.01 s2details
#220.01 s2details
#230.01 s2details
#24ACCEPTED0.01 s2details
#25--2details
#26ACCEPTED0.01 s2details
#270.01 s2details
#28--2details
#29ACCEPTED0.01 s2details
#300.01 s2details
#31ACCEPTED0.01 s1, 2details

Code

#include <bits/stdc++.h>
using namespace std;
constexpr int N = 5;
constexpr int M = 6;
constexpr int P = (6 * 6) * (6 * 6) * (6 * 6);
using T = array<int, M>;

void rot_grid(vector<string>& grid) {
	auto ret = grid;
	for (int i = 0; i < N; ++i)
		for (int j = 0; j < N; ++j)
			ret[i][j] = grid[N - j - 1][i];
	grid = ret;
}

void rot1(T& w) {
	int f = w[1];
	w[1] = w[3];
	w[3] = w[4];
	w[4] = w[5];
	w[5] = f;
}

void rot2(T& w) {
	int f = w[5];
	w[5] = w[2];
	w[2] = w[3];
	w[3] = w[0];
	w[0] = f;
}

void rot3(T& w) {
	int f = w[0];
	w[0] = w[1];
	w[1] = w[2];
	w[2] = w[4];
	w[4] = f;
}

pair<int, int> find3(vector<string>& grid) {
	for (int i = 0; i < N - 2; ++i)
		for (int j = 0; j < N; ++j)
			if (grid[i][j] != '.' && grid[i + 1][j] != '.' && grid[i + 2][j] != '.')
				return {i, j};
	return {-1, -1};
}

int get_hash(T k) {
	int r = 1, s = 0;
	for (int j = 0; j < 6; ++j, r *= 6)
		s += r * (k[j] - '0' - 1);
	assert(s < P);
	return s;
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);

	int n;
	cin >> n;

	vector<vector<int>> occ(P);
	vector<int> p;

	for (int ind = 0; ind < n; ++ind) {
		vector<string> grid(N);
		for (auto& x : grid)
			cin >> x;
		pair<int, int> l;
		l = find3(grid);
		for (int j = 0; j < 4 && l.first == -1; ++j) {
			rot_grid(grid);
			l = find3(grid);
		}
		T w {-1, -1, -1, -1, -1, -1};
		if (l.first == -1) {
			int i = 0, j = 0;
			while (grid[i][j] == '.') {
				++j;
				if (j == N) {
					i++;
					j = 0;
				}
			}
			int nb = (j + 1 < N && grid[i][j + 1] != '.') + (j - 1 >= 0 && grid[i][j - 1] != '.');
			if (nb == 0)
				rot_grid(grid);
			i = 0; j = 0;
			while (grid[i][j] == '.') {
				++j;
				if (j == N) {
					i++;
					j = 0;
				}
			}
			int oi = i, oj = j;
			vector<int> nds;
			while (i < N && j < N) {
				if (grid[i][j] != '.')
					nds.push_back(grid[i][j]);
				++j;
				if (j == N) {
					j = 0;
					i++;
				}
			}
			if (grid[oi + 1][oj] == '.') {
				w[0] = nds[4];
				w[1] = nds[0];
				w[2] = nds[1];
				w[3] = nds[2];
				w[4] = nds[3];
				w[5] = nds[5];
			} else {
				w[0] = nds[5];
				w[1] = nds[4];
				w[2] = nds[0];
				w[3] = nds[2];
				w[4] = nds[3];
				w[5] = nds[1];
			}
		} else {
			auto [i, j] = l;
			w[1] = grid[i][j];
			w[3] = grid[i + 1][j];
			w[4] = grid[i + 2][j];
			if (i + 3 < N && grid[i + 3][j] != '.') {
				w[5] = grid[i + 3][j];
				for (int kj = j + 1; kj < N; ++kj)
					for (int ki = 0; ki < N; ++ki)
						if (grid[ki][kj] != '.')
							w[w[2] == -1 ? 2 : 0] = grid[ki][kj];
				for (int kj = j - 1; kj >= 0; --kj)
					for (int ki = 0; ki < N; ++ki)
						if (grid[ki][kj] != '.')
							w[w[2] == -1 ? 2 : 0] = grid[ki][kj];
			} else {
				for (int kj = j + 1; kj < N; ++kj) {
					for (int ki = 0; ki < N; ++ki) {
						if (grid[ki][kj] != '.') {
							int dist = (i <= ki && ki <= i + 2 ? 0 : 1) + (kj - j) - 1;
							if (dist == 0)
								w[2] = grid[ki][kj];
							if (dist == 1)
								w[5] = grid[ki][kj];
							if (dist == 2)
								w[0] = grid[ki][kj];
						}
					}
				}
				for (int kj = j - 1; kj >= 0; --kj) {
					for (int ki = 0; ki < N; ++ki) {
						if (grid[ki][kj] != '.') {
							int dist = (i <= ki && ki <= i + 2 ? 0 : 1) + (j - kj) - 1;
							if (dist == 0)
								w[0] = grid[ki][kj];
							if (dist == 1)
								w[5] = grid[ki][kj];
							if (dist == 2)
								w[2] = grid[ki][kj];
						}
					}
				}
			}
		}
		if (w[1] == '1' || w[4] == '1') rot1(w);
		while (w[3] != '1') rot2(w);
		while (w[1] != (w[5] == '2' ? '3' : '2')) rot3(w);
		int x = get_hash(w);
		p.push_back(x);
		occ[x].push_back(ind);
	}

	for (int i = 0; i < n; ++i) {
		if (occ[p[i]].size() == 1) {
			cout << "-\n";
		} else {
			for (int j : occ[p[i]])
				if (i != j)
					cout << j + 1 << " ";
			cout << "\n";
		}
	}
}

Test details

Test 1

Group: 1, 2

Verdict: ACCEPTED

input
3
165..
.4...
.3...
.2...
...

correct output
3
-
1

user output

-

Test 2

Group: 1, 2

Verdict: ACCEPTED

input
5
264..
.5...
.3...
.1...
...

correct output
3
4 5
1
2 5
2 4

user output

4 5 

2 5 
2 4 

Test 3

Group: 1, 2

Verdict: ACCEPTED

input
5
152..
.4...
.3...
.6...
...

correct output
3 5
4
1 5
2
1 3

user output
3 5 

1 5 

1 3 

Test 4

Group: 1, 2

Verdict: ACCEPTED

input
5
142..
.6...
.3...
.5...
...

correct output
4 5
3
2
1 5
1 4

user output
4 5 


1 5 
1 4 

Test 5

Group: 1, 2

Verdict: ACCEPTED

input
5
123..
.4...
.6...
.5...
...

correct output
3
4 5
1
2 5
2 4

user output

4 5 

2 5 
2 4 

Test 6

Group: 1, 2

Verdict: ACCEPTED

input
5
213..
.6...
.4...
.5...
...

correct output
4 5
3
2
1 5
1 4

user output
4 5 


1 5 
1 4 

Test 7

Group: 1, 2

Verdict: ACCEPTED

input
5
314..
.5...
.2...
.6...
...

correct output
3
4 5
1
2 5
2 4

user output

4 5 

2 5 
2 4 

Test 8

Group: 1, 2

Verdict: ACCEPTED

input
5
163..
.2...
.5...
.4...
...

correct output
4 5
3
2
1 5
1 4

user output
4 5 


1 5 
1 4 

Test 9

Group: 1, 2

Verdict: ACCEPTED

input
5
264..
.1...
.3...
.5...
...

correct output
2 3
1 3
1 2
5
4

user output
2 3 
1 3 
1 2 


Test 10

Group: 1, 2

Verdict: ACCEPTED

input
5
214..
.3...
.5...
.6...
...

correct output
5
3 4
2 4
2 3
1

user output

3 4 
2 4 
2 3 

Test 11

Group: 2

Verdict:

input
10
.41..
.5...
.2...
36...
...

correct output
9
4 6 8
7 10
2 6 8
-
...

user output
(empty)

Test 12

Group: 2

Verdict:

input
10
5....
1436.
.2...
.....
...

correct output
5 8
6 10
4 7 9
3 7 9
1 8
...

user output
(empty)

Test 13

Group: 2

Verdict:

input
10
2....
41...
.63..
.5...
...

correct output
4 9 10
5 6 7 8
-
1 9 10
2 6 7 8
...

user output
(empty)

Test 14

Group: 2

Verdict: ACCEPTED

input
10
1....
634..
..52.
.....
...

correct output
2 3 4 5 6 9 10
1 3 4 5 6 9 10
1 2 4 5 6 9 10
1 2 3 5 6 9 10
1 2 3 4 6 9 10
...

user output
2 3 4 5 6 9 10 
1 3 4 5 6 9 10 
1 2 4 5 6 9 10 
1 2 3 5 6 9 10 
1 2 3 4 6 9 10 
...

Test 15

Group: 2

Verdict:

input
10
.2...
4516.
3....
.....
...

correct output
5 7 9 10
8
4 6
3 6
1 7 9 10
...

user output
(empty)

Test 16

Group: 2

Verdict: ACCEPTED

input
10
.56..
.2...
.4...
31...
...

correct output
4 9
3 5 10
2 5 10
1 9
2 3 10
...

user output
4 9 
3 5 10 
2 5 10 
1 9 
2 3 10 
...

Test 17

Group: 2

Verdict: ACCEPTED

input
10
..62.
.31..
45...
.....
...

correct output
2 3 4 8
1 3 4 8
1 2 4 8
1 2 3 8
6 7 9 10
...

user output
2 3 4 8 
1 3 4 8 
1 2 4 8 
1 2 3 8 
6 7 9 10 
...

Test 18

Group: 2

Verdict:

input
10
532..
.4...
.1...
.6...
...

correct output
3 8 9
5 6
1 8 9
7 10
2 6
...

user output
(empty)

Test 19

Group: 2

Verdict: ACCEPTED

input
10
.64..
.1...
.3...
52...
...

correct output
2 5 6 7 8 9
1 5 6 7 8 9
4 10
3 10
1 2 6 7 8 9
...

user output
2 5 6 7 8 9 
1 5 6 7 8 9 
4 10 
3 10 
1 2 6 7 8 9 
...

Test 20

Group: 2

Verdict: ACCEPTED

input
10
.4...
326..
.1...
.5...
...

correct output
4 7 8
6 9 10
5
1 7 8
3
...

user output
4 7 8 
6 9 10 

1 7 8 

...

Test 21

Group: 2

Verdict:

input
20
.6...
.4...
31...
.25..
...

correct output
3 7 11 16
6
1 7 11 16
5 19
4 19
...

user output
(empty)

Test 22

Group: 2

Verdict:

input
20
3....
5614.
..2..
.....
...

correct output
7 10 11 17 20
12
4 9 13 15 18
3 9 13 15 18
8 14 16
...

user output
7 10 11 17 20 
12 
4 9 13 18 
3 9 13 18 
8 14 16 
...

Test 23

Group: 2

Verdict:

input
20
42...
.316.
.5...
.....
...

correct output
5 12 13 15 18
16 20
6 8 14
9 19
1 12 13 15 18
...

user output
12 13 15 18 
16 20 
6 8 14 
9 19 
-
...

Test 24

Group: 2

Verdict: ACCEPTED

input
20
..5..
.623.
41...
.....
...

correct output
2 6 11 12 13
1 6 11 12 13
5 16 18
7 14
3 16 18
...

user output
2 6 11 12 13 
1 6 11 12 13 
5 16 18 
7 14 
3 16 18 
...

Test 25

Group: 2

Verdict:

input
20
.46..
53...
.1...
.2...
...

correct output
2 3 5 7 15 17 19
1 3 5 7 15 17 19
1 2 5 7 15 17 19
8 10 11 14
1 2 3 7 15 17 19
...

user output
(empty)

Test 26

Group: 2

Verdict: ACCEPTED

input
20
.61..
.4...
35...
.2...
...

correct output
8 10 20
3 17 18 19
2 17 18 19
14 15
6 7 9 13
...

user output
8 10 20 
3 17 18 19 
2 17 18 19 
14 15 
6 7 9 13 
...

Test 27

Group: 2

Verdict:

input
20
..2..
1463.
.5...
.....
...

correct output
2 3 5 6 9 20
1 3 5 6 9 20
1 2 5 6 9 20
11 19
1 2 3 6 9 20
...

user output
2 3 5 6 9 20 
1 3 5 6 9 20 
1 2 5 6 9 20 
19 
1 2 3 6 9 20 
...

Test 28

Group: 2

Verdict:

input
20
...4.
5132.
6....
.....
...

correct output
2 8 10 12 13 19
1 8 10 12 13 19
4 5 15 16 17
3 5 15 16 17
3 4 15 16 17
...

user output
(empty)

Test 29

Group: 2

Verdict: ACCEPTED

input
20
.2...
.31..
45...
6....
...

correct output
5 8 9 14 17
3 10 16
2 10 16
13 15 19
1 8 9 14 17
...

user output
5 8 9 14 17 
3 10 16 
2 10 16 
13 15 19 
1 8 9 14 17 
...

Test 30

Group: 2

Verdict:

input
20
3....
452..
.1...
.6...
...

correct output
3 7 8 9 14 15 16 19
4 12 13 17
1 7 8 9 14 15 16 19
2 12 13 17
11 20
...

user output
3 7 8 9 14 15 16 
4 12 13 17 
1 7 8 9 14 15 16 
2 12 13 17 
11 20 
...

Test 31

Group: 1, 2

Verdict: ACCEPTED

input
2
546..
.3...
.2...
.1...
...

correct output
-
-

user output
-
-