Ruudukko

#!/usr/bin/env python3

TILAN_MAX_LEV = 18
TILAN_MAX_KORK = 5 # (5 + väli) per tila

# ruudukko koostuu yhdestä 1-20 reitin tilasta
# ja mahdollisesti yhdestä tai kahdesta isommasta reittitilasta

def tilan_reitit(x, y):
    tila = [[1 for a in range(x)] for b in range(y)]
    for b in range(1, y):
        for a in range(1, x):
            tila[b][a] = tila[b-1][a] + tila[b][a-1]
    return tila

def sopiva_ruudukko(kohdemaara, tilaruudukko):
    paras = 0
    valinta = None
    for y in range(TILAN_MAX_KORK):
        for x in range(TILAN_MAX_LEV):
            if paras < tilaruudukko[y][x] <= kohdemaara:
                paras = tilaruudukko[y][x]
                valinta = (x + 1, y + 1)
    return valinta, paras

def maarita_ruudukot(reitteja):
    tilaruudukko = tilan_reitit(TILAN_MAX_LEV, TILAN_MAX_KORK)
    ruudukot = []
    while reitteja > 20:
        valinta, kattaa = sopiva_ruudukko(reitteja, tilaruudukko)
        reitteja -= kattaa
        ruudukot.append(valinta)
    return (ruudukot, reitteja)

def luo_kentta(ruudukot, jaljella):
    rivit = []
    if jaljella > 0:
        rivit.append('.' * jaljella + '#' * (20 - jaljella))
        rivit.append('.' * 20)

    for x, y in ruudukot:
        rivit.append('.' + 18 * '#' + '.')
        for j in range(y - 1):
            rivit.append('.' * x + '#' * (19 - x) + '.')
        rivit.append('.' * 20)

    while len(rivit) < 20:
        rivit.append(19 * '#' + '.')
    return rivit

def piirra_kentta(kentta):
    for rivi in kentta:
        print(rivi)

def main():
    n = int(input())
    ruudukot, jaljella = maarita_ruudukot(n)
    kentta = luo_kentta(ruudukot, jaljella)
    piirra_kentta(kentta)

if __name__ == '__main__':
    main()

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: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED

Test 10

Group: 2

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: ACCEPTED

Test 15

Group: 3

Verdict: ACCEPTED