1135 - p2sah

Se dă o tablă de șah cu n+1 linii (numerotate de sus în jos începând cu 1) și 2n+1 coloane (numerotate de la stânga la dreapta începând cu 1). Pe prima linie pătratul din mijloc conține 1 gram de fân, iar celelalte pătrate de pe prima linie nu conțin nimic. Începând cu linia a doua fiecare pătrat conține o cantitate de fân obținută prin adunarea cantităților de fân din cele 3 pătrate ale liniei anterioare cu care se învecinează (pe verticală și diagonală). De exemplu dacă n=3 tabla are 4 linii, 7 coloane și următoarea configurație.
Un cal pleacă de pe prima linie, de pe o coloana k<=n, sare din orice poziție (i,j) în poziția (i+1,j+2) atât timp cât este posibil și mănâncă tot fânul din pătratele prin care trece. De exemplu, pentru n=3 și k=2, pătratele prin care trece calul sunt marcate cu asterisc (*)

Cerința

1. Cunoscând n și k, să se calculeze cantitatea de fân de pe linia k a tablei.
2. Cunoscând n și k, să se calculeze câte grame de fân mănâncă un cal care pleacă de pe prima linie, de pe coloana k.
Întrucât aceste numere pot fi mari, se cere doar restul modulo 100003 ale acestor numere.

Date de intrare

Fișierul de intrare p2sahin.txt conține pe prima linie un număr t cu valoarea 1 sau 2. Pe a doua linie a fișierului de intrare se găsesc două numere naturale n și k separate printr-un spațiu.
Dacă t=1 se va rezolva prima cerință, deci pentru valoarea n citită tabla are n+1 linii și 2n+1 coloane, iar k reprezintă numărul liniei de pe care trebuie calculată cantitatea de fân.
Dacă t=2 se va rezolva a doua cerință, deci pentru valoarea n citită tabla are n+1 linii și 2n+1 coloane, iar k reprezintă numărul coloanei din prima linie de unde pleacă calul.

Date de ieșire

Dacă t din fișierul de intrare este 1 se va rezolva doar prima cerință.
În acest caz fișierul de ieșire p2sahout.txt va conține un singur număr reprezentând cantitatea totală de fân din toate pătratele situate pe tabla pe linia k (trebuie afișat restul modulo 100003).
Dacă t din fișierul de intrare este 2 se va rezolva doar a doua cerință.
În acest caz fișierul de ieșire p2sahout.txt va conține un singur număr reprezentând cantitatea totală de fân mâncată de un cal care pleacă de pe linia 1 și coloana k (trebuie afișat restul modulo 100003).

Restricții și precizări

• 1 ≤ k ≤ n ≤ 1.000.000.000
• La cerința 1 pentru 80% dintre teste k ≤ n ≤ 1.000.000, iar pentru alte 20% din teste k ≤ n ≤ 1.000.000.000
• La cerința 2 pentru 30% dintre teste k ≤ n ≤ 1000, pentru alte 30% dintre teste k ≤ n ≤ 1.000.000, iar pentru restul de 40% dintre teste k ≤ n ≤ 1.000.000.000.
• Rezolvarea corectă a primei cerințe asigură 20% din punctajul testului respectiv.
• Rezolvarea corectă a celei de a doua cerințe asigura 70% din punctajul testului respectiv.

Exemplul 1

p2sahin.txt

1

3 2

p2sahout.txt

3

Explicatie

t=1, deci se rezolvă prima cerință.
Pe linia a doua există 3 pătrate care conțin fiecare câte un gram de fân.(vezi desenul din enunț)

Exemplul 2

p2sahin.txt

2

3 2

p2sahout.txt

2

Explicatie

t=2, deci se rezolvă a doua cerință.
Traseul calului este: (1,2) -> (2,4) -> (3,6) adică exact pătrățelele marcate cu asterisc în desenul din enunț. Prima poziție nu conține fân, iar celelalte două conțin câte un gram de fân. Deci calul mănâncă 2 grame de fân.

Rezolvare

#1135 - p2sah
def calculate_forage_line(n, k):
    MOD = 100003

    # Inițializăm prima linie a tablei
    table = [[0] * (2 * n + 1) for _ in range(n + 1)]
    table[0][n] = 1

    # Construim tabela în conformitate cu regulile date
    for i in range(1, n + 1):
        for j in range(1, 2 * n):
            table[i][j] = (table[i - 1][j - 1] + table[i - 1][j] + table[i - 1][j + 1]) % MOD

    # Calculăm suma pentru linia k
    total_forage = sum(table[k]) % MOD

    return total_forage

# Citirea din fișier
with open("p2sahin.txt", "r") as file:
    t = int(file.readline())
    n, k = map(int, file.readline().split())

# Rezolvarea cerinței 1
if t == 1:
    result = calculate_forage_line(n, k)

    # Scrierea în fișier
    with open("p2sahout.txt", "w") as output_file:
        output_file.write(str(result))
def calculate_horse_forage(n, k):
    MOD = 100003

    # Inițializăm prima linie a tablei
    table = [[0] * (2 * n + 1) for _ in range(n + 1)]
    table[0][n] = 1

    # Construim tabela în conformitate cu regulile date
    for i in range(1, n + 1):
        for j in range(1, 2 * n):
            table[i][j] = (table[i - 1][j - 1] + table[i - 1][j] + table[i - 1][j + 1]) % MOD

    # Calculăm suma pentru cal
    total_forage = 0
    current_row = k
    current_col = n

    while current_row <= n:
        total_forage = (total_forage + table[current_row][current_col]) % MOD
        current_row += 1
        current_col += 2

    return total_forage

# Citirea din fișier
with open("p2sahin.txt", "r") as file:
    t = int(file.readline())
    n, k = map(int, file.readline().split())

# Rezolvarea cerinței 2
if t == 2:
    result = calculate_horse_forage(n, k)

    # Scrierea în fișier
    with open("p2sahout.txt", "w") as output_file:
        output_file.write(str(result))