CSES - NOI 2024 - Results
Submission details
Task:Chair Game
Sender:Ruiming Zhang
Submission time:2024-03-06 17:31:02 +0200
Language:CPython3
Status:READY
Result:5
Feedback
groupverdictscore
#10
#2ACCEPTED5
#30
#40
#50
#60
#70
#80
Test results
testverdicttimegroup
#1ACCEPTED0.94 s1, 7, 8details
#2ACCEPTED0.07 s1, 7, 8details
#3ACCEPTED0.79 s1, 7, 8details
#4--1, 7, 8details
#5ACCEPTED0.97 s1, 7, 8details
#60.02 s7, 8details
#70.02 s7, 8details
#8ACCEPTED0.05 s2, 8details
#9--3, 4, 5, 6, 8details
#10ACCEPTED0.02 s3, 4, 5, 6, 8details
#11ACCEPTED0.04 s3, 4, 5, 6, 8details
#12ACCEPTED0.07 s3, 4, 5, 6, 8details
#13--4, 5, 6, 7, 8details
#140.05 s4, 5, 6, 8details
#150.06 s4, 5, 6, 8details
#160.06 s4, 5, 6, 8details
#170.02 s5, 6, 7, 8details
#180.02 s5, 6, 8details
#190.02 s5, 6, 8details
#200.02 s5, 6, 8details
#21--1, 6, 7, 8details
#220.02 s6, 7, 8details
#230.02 s6, 8details
#240.02 s6, 8details
#250.02 s8details
#260.02 s8details
#27ACCEPTED0.05 s3, 4, 5, 6, 8details
#28ACCEPTED0.05 s8details
#290.02 s8details
#300.02 s8details

Code

from itertools import permutations

t = int(input())

for _ in range(t):
    n = int(input())
    lst = list(map(int, input().split()))

    if len(set(lst)) == 1:
        print("YES")
        print(*lst)
    elif len(set(lst)) == n:
        if n % 2:
            print("YES")
            print(n, end = " ")
            for i in range(1, n):
                print(i, end = " ")
            print()
        else:
            print("NO")
    else:
        if n <= 8:
            all = permutations(lst)

            t = True
            for i in all:
                p = True
                res = [0] * n
                for j in range(n):
                    temp = j + i[j]
                    while temp >= n:
                        temp -= n
                    res[temp] += 1

                for k in res:
                    if k == 0 or k > 1:
                        p = False
                        break
                if p:
                    print("YES")
                    print(*i)
                    t = False
                    break
            if t:
                print("NO")
                    
        else:
            if len(set(lst)) == 1:
                print("YES")
                for i in lst:
                    print(i, end = " ")
                print()
            else:
                if set(lst) == {1, 2} or set(lst) == {3, 1} or set(lst) == {3, 2}:
                    print("NO")
                else:
                    dic = {1:0, 2:0, 3:0}
                    for i in lst:
                        dic[i] += 1

                    if dic[1] == dic[2] and dic[2] == dic[3]:
                        print("YES")
                        for i in range(dic[1]):
                            print("3 1 2", end = " ")
                        print()
                    elif dic[1] == dic[3] and dic[2] == dic[3] - 1:
                        print("YES")
                        for i in range(dic[2]):
                            print("3 1 2", end = " ")
                        print("3 1")

                    else:
                        print("NO")

Test details

Test 1

Group: 1, 7, 8

Verdict: ACCEPTED

input
637
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES
1
YES
1 1
YES
...
Truncated

Test 2

Group: 1, 7, 8

Verdict: ACCEPTED

input
246
7
1 1 1 1 1 1 1
7
1 1 2 1 1 7 1
...

correct output
YES
1 1 1 1 1 1 1 
YES
1 1 1 1 2 7 1 
YES
...

user output
YES
1 1 1 1 1 1 1
YES
1 1 2 7 1 1 1
YES
...
Truncated

Test 3

Group: 1, 7, 8

Verdict: ACCEPTED

input
810
8
1 1 1 1 1 1 1 1
8
1 1 1 8 1 1 2 1
...

correct output
YES
1 1 1 1 1 1 1 1 
YES
1 1 2 8 1 1 1 1 
YES
...

user output
YES
1 1 1 1 1 1 1 1
YES
1 1 1 1 1 2 8 1
YES
...
Truncated

Test 4

Group: 1, 7, 8

Verdict:

input
1000
8
8 8 5 2 8 7 6 5
8
6 5 2 2 8 2 1 6
...

correct output
NO
YES
8 2 2 6 2 5 1 6 
NO
NO
...

user output
(empty)

Test 5

Group: 1, 7, 8

Verdict: ACCEPTED

input
1000
8
2 1 7 7 2 3 8 2
8
4 1 5 4 7 3 5 3
...

correct output
YES
7 2 2 7 1 3 8 2 
YES
4 4 7 3 3 5 5 1 
YES
...

user output
YES
2 7 1 2 3 7 8 2
YES
4 1 5 3 5 3 7 4
YES
...
Truncated

Test 6

Group: 7, 8

Verdict:

input
1000
16
15 16 6 4 14 2 1 6 2 16 10 2 9...

correct output
NO
NO
NO
NO
NO
...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 7

Group: 7, 8

Verdict:

input
1000
16
2 4 13 6 8 16 12 8 16 5 9 5 9 ...

correct output
YES
13 5 2 8 12 2 8 5 16 16 9 6 9 ...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 8

Group: 2, 8

Verdict: ACCEPTED

input
1000
1
1
2
1 2
...

correct output
YES

NO
YES
3 1 2 
...

user output
YES
1
NO
YES
3 1 2 
...
Truncated

Test 9

Group: 3, 4, 5, 6, 8

Verdict:

input
988
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
(empty)

Test 10

Group: 3, 4, 5, 6, 8

Verdict: ACCEPTED

input
199
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES
1
YES
1 1
YES
...
Truncated

Test 11

Group: 3, 4, 5, 6, 8

Verdict: ACCEPTED

input
1000
100
1 1 1 2 1 1 2 2 1 1 1 1 1 2 1 ...

correct output
NO
NO
NO
NO
NO
...

user output
NO
NO
NO
NO
NO
...
Truncated

Test 12

Group: 3, 4, 5, 6, 8

Verdict: ACCEPTED

input
1000
100
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
YES
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
YES
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...
Truncated

Test 13

Group: 4, 5, 6, 7, 8

Verdict:

input
963
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
(empty)

Test 14

Group: 4, 5, 6, 8

Verdict:

input
979
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES
1
YES
1 1
YES
...
Truncated

Test 15

Group: 4, 5, 6, 8

Verdict:

input
1000
100
3 3 1 2 1 1 2 3 1 3 2 1 1 3 1 ...

correct output
NO
NO
NO
NO
NO
...

user output
NO
NO
NO
NO
NO
...
Truncated

Test 16

Group: 4, 5, 6, 8

Verdict:

input
1000
100
1 2 2 2 2 1 1 1 2 3 1 1 3 2 1 ...

correct output
YES
2 2 2 3 1 2 3 1 2 3 1 3 1 3 1 ...

user output
NO
NO
NO
NO
NO
...
Truncated

Test 17

Group: 5, 6, 7, 8

Verdict:

input
980
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES
1
YES
1 1
YES
...
Truncated

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 18

Group: 5, 6, 8

Verdict:

input
947
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES
1
YES
1 1
YES
...
Truncated

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 19

Group: 5, 6, 8

Verdict:

input
1000
100
1 2 4 2 1 3 1 2 2 3 1 1 3 1 4 ...

correct output
NO
NO
NO
NO
NO
...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 20

Group: 5, 6, 8

Verdict:

input
1000
100
3 4 4 4 4 4 4 3 3 3 4 4 2 3 3 ...

correct output
YES
4 2 4 4 1 3 4 2 4 2 3 4 2 4 4 ...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 21

Group: 1, 6, 7, 8

Verdict:

input
715
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
(empty)

Test 22

Group: 6, 7, 8

Verdict:

input
843
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES
1
YES
1 1
YES
...
Truncated

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 23

Group: 6, 8

Verdict:

input
1000
100
3 4 5 1 4 4 2 3 2 3 4 1 1 1 2 ...

correct output
NO
NO
NO
NO
NO
...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 24

Group: 6, 8

Verdict:

input
1000
100
5 3 4 3 5 3 3 5 5 4 5 5 5 5 2 ...

correct output
YES
4 4 5 5 2 4 4 5 3 5 5 2 5 5 2 ...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 25

Group: 8

Verdict:

input
1000
100
88 70 59 44 28 10 19 19 42 16 ...

correct output
NO
NO
NO
NO
NO
...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 26

Group: 8

Verdict:

input
1000
100
31 72 52 30 77 56 79 10 88 11 ...

correct output
YES
31 62 14 10 66 63 1 82 37 92 3...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 27

Group: 3, 4, 5, 6, 8

Verdict: ACCEPTED

input
1000
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES
1
YES
1 1
YES
...
Truncated

Test 28

Group: 8

Verdict: ACCEPTED

input
1000
1
1
2
2 2
...

correct output
YES

YES
2 2 
YES
...

user output
YES
1
YES
2 2
YES
...
Truncated

Test 29

Group: 8

Verdict:

input
1000
100
87 81 29 35 8 98 77 50 46 34 5...

correct output
YES
34 74 25 91 80 18 95 26 88 12 ...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...

Test 30

Group: 8

Verdict:

input
1000
100
65 92 39 22 67 41 17 65 97 71 ...

correct output
YES
9 38 24 59 69 24 63 3 22 35 24...

user output
(empty)

Error:
Traceback (most recent call last):
  File "/box/input/code.py", line 59, in <module>
    d...