Kirjat

import itertools

def getUniqPerm(fNums, sNums, force):
    if force:
        return getPerm(fNums, sNums) # käytetään kaikki luvut
    else:
        fTriplet = fNums[0:3]
        sTriplet = sNums[0:3]
        if hasSolution(fTriplet, sTriplet):
            return getPerm(fTriplet, sTriplet) # jos ensimmäisellä kolmikolla on ratkaisu, selvitetään se
        return getPerm(fNums, sNums) # jos ei ratkaisua, etsitään ratkaisu kaikille pareille
        

# vajaa testi, palauttaa vain tiedon, ovatko kaksi lukuparia eri
def pairEquals(x1, x2, y1, y2):
    if x1 != y1 and x1 != y2:
        return False
    return True
    
        
# onko kolmella parilla ratkaisua vai tarvitaanko neljä
def hasSolution(fNums, sNums):
    if pairEquals(fNums[0], fNums[1], sNums[0], sNums[1]):
        return False
    if pairEquals(fNums[1], fNums[2], sNums[1], sNums[2]):
        return False
    if pairEquals(fNums[0], fNums[2], sNums[0], sNums[2]):
        return False
    return True    
    
def getPerm(fNums, sNums):
    # print ('fNums length {}'.format(len(fNums)))
    # print ('fNums {}'.format(fNums))
    # possibilities = [] # fNums ja sNums indeksin eli päivän kohdalle mahdolliset kirjat
    # books = [False] * len(fNums) # False = lukematon, True = luettu
    # solution = [-1] * len(fNums) # Mikä kirja minäkin päivänä
    # 
    # for i in range(0, len(fNums)):
    #     singlePossibilities = []
    #     for j in fNums:
    #         if j != (fNums[i]) and j != (sNums[i]):
    #             singlePossibilities.append(j)
    #     possibilities.append(singlePossibilities)
    # 
    # 
    # 
    # print (possibilities)
    # for i in range(0, len(fNums)):
    #     for j in range(0, len(possibilities[i])):
    #         if books[possibilities[i][j]] == False:
    #             solution[i] = possibilities[i][j] + 1 
    #             books[possibilities[i][j]] = True
    #             break
    # return solution
    
    perms = list(itertools.permutations(fNums))  # O(n!), mutta 3<=n<=7, keskimäärin 3<=n<=4
    
    for i in range(0, len(perms)):
        found = True
        for j in range(0, len(perms[i])):
            if perms[i][j] == fNums[j] or perms[i][j] == sNums[j]:
                found = False
                break
        if found:
            return list(perms[i])
    
    return [-1]


bookCount = int(input())

fSchedule = input()
fSchedule = [int(x) for x in fSchedule.split(' ')]

sSchedule = input()
sSchedule = [int(x) for x in sSchedule.split(' ')]

result = ''

for i in range(0, bookCount, 3):
    if (len(fSchedule) - i < 8):       
        elems = len(fSchedule) - i
        resRaw = getUniqPerm(fSchedule[i:i+elems], sSchedule[i:i+elems], True)
        for i in resRaw:
            result += str(i) + ' '
        break
    else:
        resRaw = getUniqPerm(fSchedule[i:i+4], sSchedule[i:i+4], False)
        for i in resRaw:
            result += str(i) + ' '
        if len(resRaw) == 4:
            i += 1
        
print (result)

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 1

Verdict: ACCEPTED

Test 7

Group: 1

Verdict: ACCEPTED

Test 8

Group: 1

Verdict: ACCEPTED

Test 9

Group: 1

Verdict: ACCEPTED

Test 10

Group: 1

Verdict: ACCEPTED

Test 11

Group: 1

Verdict: ACCEPTED

Test 12

Group: 1

Verdict: ACCEPTED

Test 13

Group: 1

Verdict: ACCEPTED

Test 14

Group: 1

Verdict: ACCEPTED

Test 15

Group: 1

Verdict: ACCEPTED

Test 16

Group: 1

Verdict: ACCEPTED

Test 17

Group: 1

Verdict: ACCEPTED

Test 18

Group: 1

Verdict: WRONG ANSWER

Test 19

Group: 1

Verdict: ACCEPTED

Test 20

Group: 1

Verdict: WRONG ANSWER

Test 21

Group: 2

Verdict: ACCEPTED

Test 22

Group: 2

Verdict: ACCEPTED

Test 23

Group: 2

Verdict: ACCEPTED

Test 24

Group: 2

Verdict: ACCEPTED

Test 25

Group: 2

Verdict: ACCEPTED

Test 26

Group: 2

Verdict: ACCEPTED

Test 27

Group: 2

Verdict: ACCEPTED

Test 28

Group: 2

Verdict: ACCEPTED

Test 29

Group: 2

Verdict: ACCEPTED

Test 30

Group: 2

Verdict: ACCEPTED

Test 31

Group: 2

Verdict: ACCEPTED

Test 32

Group: 2

Verdict: ACCEPTED

Test 33

Group: 2

Verdict: ACCEPTED

Test 34

Group: 2

Verdict: ACCEPTED

Test 35

Group: 2

Verdict: ACCEPTED

Test 36

Group: 2

Verdict: ACCEPTED

Test 37

Group: 2

Verdict: WRONG ANSWER

Test 38

Group: 2

Verdict: WRONG ANSWER

Test 39

Group: 2

Verdict: ACCEPTED

Test 40

Group: 2

Verdict: WRONG ANSWER

Test 41

Group: 3

Verdict: ACCEPTED

Test 42

Group: 3

Verdict: ACCEPTED

Test 43

Group: 3

Verdict: ACCEPTED

Test 44

Group: 3

Verdict: ACCEPTED

Test 45

Group: 3

Verdict: ACCEPTED

Test 46

Group: 3

Verdict: ACCEPTED

Test 47

Group: 3

Verdict: ACCEPTED

Test 48

Group: 3

Verdict: ACCEPTED

Test 49

Group: 3

Verdict: ACCEPTED

Test 50

Group: 3

Verdict: ACCEPTED

Test 51

Group: 3

Verdict: ACCEPTED

Test 52

Group: 3

Verdict: ACCEPTED

Test 53

Group: 3

Verdict: ACCEPTED

Test 54

Group: 3

Verdict: ACCEPTED

Test 55

Group: 3

Verdict: ACCEPTED

Test 56

Group: 3

Verdict: WRONG ANSWER

Test 57

Group: 3

Verdict: ACCEPTED

Test 58

Group: 3

Verdict: WRONG ANSWER

Test 59

Group: 3

Verdict: ACCEPTED

Test 60

Group: 3

Verdict: ACCEPTED