CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit I
Sender:DiscoFlower8890
Submission time:2024-11-05 12:30:01 +0200
Language:Python3 (PyPy3)
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED12
#2ACCEPTED15
#3ACCEPTED73
Test results
testverdicttimegroup
#1ACCEPTED0.04 s1, 2, 3details
#2ACCEPTED0.07 s2, 3details
#3ACCEPTED0.15 s3details
#4ACCEPTED0.14 s3details
#5ACCEPTED0.13 s3details
#6ACCEPTED0.15 s3details
#7ACCEPTED0.11 s3details
#8ACCEPTED0.12 s3details
#9ACCEPTED0.12 s3details
#10ACCEPTED0.13 s3details
#11ACCEPTED0.13 s3details
#12ACCEPTED0.13 s3details
#13ACCEPTED0.14 s3details
#14ACCEPTED0.14 s3details
#15ACCEPTED0.15 s3details
#16ACCEPTED0.16 s3details
#17ACCEPTED0.16 s3details
#18ACCEPTED0.15 s3details
#19ACCEPTED0.15 s3details
#20ACCEPTED0.16 s3details

Code

round_index = 0
player1_sequence = []
player2_sequence = []
p1_remaining_cards = []
p2_remaining_cards = []

def cardsA(j):
    global player2_sequence, player1_sequence, p1_remaining_cards, p2_remaining_cards, round_index
    for _ in range(j):
        for i in range(len(p2_remaining_cards)):
            found = False
            if p1_remaining_cards[-i] > p2_remaining_cards[0]:
                player1_sequence[round_index] = p1_remaining_cards.pop(-i)
                player2_sequence[round_index] = p2_remaining_cards.pop(0)
                round_index += 1
                found = True
                break
        if not found:
            raise IndexError("Couldn't find a card pair")

def cardsB(j):
    global player2_sequence, player1_sequence, p1_remaining_cards, p2_remaining_cards, round_index
    for _ in range(j):
        for i in range(len(p1_remaining_cards)):
            found = False
            if p2_remaining_cards[-i] > p1_remaining_cards[0]:
                player2_sequence[round_index] = p2_remaining_cards.pop(-i)
                player1_sequence[round_index] = p1_remaining_cards.pop(0)
                round_index += 1
                found = True
                break
        if not found:
            raise IndexError("Couldn't find a card pair")

def generate_game(n, a, b):
    if a + b > n or n == 0:
        raise IndexError("Just no.")

    global player1_sequence, player2_sequence, p1_remaining_cards, p2_remaining_cards, round_index
    
    player1_sequence = ["0"] * n
    player2_sequence = ["0"] * n

    available_cards = list(range(1, n + 1))
    round_index = 0
    draws = n - a - b
    p1_remaining_cards = available_cards
    p2_remaining_cards = available_cards

    # Round which end in a draw
    middle_index_start = (n - draws) // 2
    middle_cards = available_cards[middle_index_start:middle_index_start + draws]
    
    # taking the cards for the draws from the middle of the deck so both players can still win and lose
    p1_remaining_cards = [card for card in available_cards if card not in middle_cards]
    p2_remaining_cards = [card for card in available_cards if card not in middle_cards]
    
    for card in middle_cards:
        player1_sequence[round_index] = card
        player2_sequence[round_index] = card
        round_index += 1

    # starting with the smaller amount of wins
    if a < b:
        cardsA(a)
        cardsB(b)
    else:
        cardsB(b)
        cardsA(a)

    return player1_sequence, player2_sequence

def main():
    t = int(input())
    for _ in range(t):
        n, a, b = map(int, input().strip().split())
        try:
            p1_cards, p2_cards  = generate_game(n, a, b)

            print("YES")
            print(*p1_cards, sep=' ')
            print(*p2_cards, sep=' ')
        except IndexError:
            print("NO")
    return

if __name__ == "__main__":
        main()

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
54
4 4 0
3 1 3
3 2 2
4 0 4
...

correct output
NO
NO
NO
NO
NO
...

user output
NO
NO
NO
NO
NO
...

Test 2

Group: 2, 3

Verdict: ACCEPTED

input
284
6 1 0
5 0 2
7 1 5
7 7 5
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 
2 3 4 5 6 1 7 
...

user output
NO
NO
YES
4 7 1 2 3 5 6
4 1 2 3 5 6 7
...

Test 3

Group: 3

Verdict: ACCEPTED

input
955
14 2 10
12 2 5
10 4 9
14 1 13
...

correct output
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
YES
7 8 14 13 1 2 3 4 5 6 9 10 11 ...

Test 4

Group: 3

Verdict: ACCEPTED

input
869
17 12 9
16 8 4
15 9 9
17 11 15
...

correct output
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
NO
YES
7 8 9 10 1 2 3 4 5 6 11 12 13 ...

Test 5

Group: 3

Verdict: ACCEPTED

input
761
18 3 15
19 1 15
18 8 1
19 19 17
...

correct output
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
YES
18 17 16 1 2 3 4 5 6 7 8 9 10 ...

Test 6

Group: 3

Verdict: ACCEPTED

input
925
21 14 21
20 18 18
20 7 6
21 14 9
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
NO
NO
YES
7 8 9 10 11 12 13 1 2 3 4 5 6 ...

Test 7

Group: 3

Verdict: ACCEPTED

input
529
22 3 3
22 17 5
22 6 15
22 22 20
...

correct output
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
YES
4 5 6 7 8 9 10 11 12 13 14 15 ...

Test 8

Group: 3

Verdict: ACCEPTED

input
576
23 18 9
23 16 8
23 16 13
23 16 22
...

correct output
NO
NO
NO
NO
NO
...

user output
NO
NO
NO
NO
NO
...

Test 9

Group: 3

Verdict: ACCEPTED

input
625
24 2 22
24 15 21
24 6 3
24 21 1
...

correct output
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
YES
24 23 1 2 3 4 5 6 7 8 9 10 11 ...

Test 10

Group: 3

Verdict: ACCEPTED

input
676
25 16 25
25 15 2
25 15 7
25 15 16
...

correct output
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
NO
YES
9 10 11 12 13 14 15 16 1 2 3 4...

Test 11

Group: 3

Verdict: ACCEPTED

input
729
26 2 18
26 14 18
26 5 18
26 19 13
...

correct output
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
YES
11 12 13 14 15 16 26 25 1 2 3 ...

Test 12

Group: 3

Verdict: ACCEPTED

input
784
27 26 7
27 14 0
27 14 5
27 14 14
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
NO
NO
YES
10 11 12 13 14 15 16 17 1 2 3 ...

Test 13

Group: 3

Verdict: ACCEPTED

input
841
28 26 16
28 13 19
28 5 8
28 26 4
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
NO
NO
YES
7 8 9 10 11 12 13 14 15 16 17 ...

Test 14

Group: 3

Verdict: ACCEPTED

input
900
29 24 15
29 13 2
29 13 7
29 13 16
...

correct output
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
NO
YES
8 9 10 11 12 13 14 15 16 17 18...

Test 15

Group: 3

Verdict: ACCEPTED

input
961
30 24 26
30 12 24
30 4 29
30 24 14
...

correct output
NO
NO
NO
NO
YES
...

user output
NO
NO
NO
NO
YES
...

Test 16

Group: 3

Verdict: ACCEPTED

input
1000
15 12 6
33 18 30
44 4 26
6 6 5
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
NO
NO
YES
16 17 18 19 20 21 22 23 24 25 ...

Test 17

Group: 3

Verdict: ACCEPTED

input
1000
45 32 30
4 0 3
46 23 10
71 19 46
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
NO
NO
YES
17 18 19 20 21 22 23 24 25 26 ...

Test 18

Group: 3

Verdict: ACCEPTED

input
1000
51 29 37
75 11 72
5 2 4
31 8 26
...

correct output
NO
NO
NO
NO
YES
...

user output
NO
NO
NO
NO
YES
...

Test 19

Group: 3

Verdict: ACCEPTED

input
1000
50 20 37
99 45 58
86 79 73
85 70 54
...

correct output
NO
NO
NO
NO
NO
...

user output
NO
NO
NO
NO
NO
...

Test 20

Group: 3

Verdict: ACCEPTED

input
1000
26 23 5
73 53 59
64 47 41
80 75 55
...

correct output
NO
NO
NO
NO
NO
...

user output
NO
NO
NO
NO
NO
...