Kortit I

import cProfile, random

def generate_game(n, a, b):
    if a + b > n:
        raise IndexError("Too many wins compared to rounds.")
    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

    for _ in range(draws):
        # 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

    # Wins for player A
    for _ in range(a):
        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")

    # Wins for player B
    for _ in range(b):
        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")
    
    # check if we slipped a few zeros in the list
    #TODO Wait shouldn't the wins part pick that up?
    check1 = set(player1_sequence)
    check2 = set(player2_sequence)

    if 0 in check1 or 0 in check2:
        raise IndexError("Left a few 0's game is not possible")

    return player1_sequence, player2_sequence

def main(data):
    #t = int(input())
    for i in range(len(data)):
        #n, a, b = map(int, input().strip().split())
        try:
            #p1_cards, p2_cards = generate_game(3, 3, 0)
            #p1_cards, p2_cards  = generate_game(n, a, b)
            p1_cards, p2_cards  = generate_game(data[i][0], data[i][1], data[i][2])
            print(f"YES\n{p1_cards}\n{p2_cards}")
            print(data[i])
            print()
        except IndexError:
            print("NO")
            print(data[i])
            print()
    return

# Test case generation and profiling
def generate_test_cases_checked(num_tests):
    test_cases = []
    for _ in range(num_tests):
        n = random.randint(1, 101)
        while True:
            a = random.randint(0, n)
            b = random.randint(0, n)
            if a + b <= n:
                test_cases.append((n, a, b))
                break
    return test_cases

# Run with profiling
test_cases = generate_test_cases_checked(1000)
cProfile.run('main(test_cases)')

#if __name__ == "__main__":
#    main(test_cases)

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 2, 3

Verdict: WRONG ANSWER

Test 3

Group: 3

Verdict: WRONG ANSWER

Test 4

Group: 3

Verdict: WRONG ANSWER

Test 5

Group: 3

Verdict: WRONG ANSWER

Test 6

Group: 3

Verdict: WRONG ANSWER

Test 7

Group: 3

Verdict: WRONG ANSWER

Test 8

Group: 3

Verdict: WRONG ANSWER

Test 9

Group: 3

Verdict: WRONG ANSWER

Test 10

Group: 3

Verdict: WRONG ANSWER

Test 11

Group: 3

Verdict: WRONG ANSWER

Test 12

Group: 3

Verdict: WRONG ANSWER

Test 13

Group: 3

Verdict: WRONG ANSWER

Test 14

Group: 3

Verdict: WRONG ANSWER

Test 15

Group: 3

Verdict: WRONG ANSWER

Test 16

Group: 3

Verdict: WRONG ANSWER

Test 17

Group: 3

Verdict: WRONG ANSWER

Test 18

Group: 3

Verdict: WRONG ANSWER

Test 19

Group: 3

Verdict: WRONG ANSWER

Test 20

Group: 3

Verdict: WRONG ANSWER