1584 - Spirala2

Cerința

Se consideră un tablou bidimensional format din n linii și n coloane, completat cu elemente de la 1 la n2. Completarea acestuia se face pornind din colțul din stânga sus la cel din dreapta sus, de la cel din dreapta sus la cel din dreapta jos ș.a. (exemplu mai jos).

Scrieți un program care citește un număr natural n și determină:

1. Suma resturilor împărțirii la 100003 a elementelor de pe linia k și de pe coloana k; elementul care se află la intersecția acestora nu se va adăuga în sumă.
2. Matricea formată prin inversarea coloanei k cu linia k.

Date de intrare

Fișierul de intrare spirala2.in conține pe prima linie un număr natural p. Pentru toate testele de intrare, numărul p poate avea doar valoarea 1 sau 2. A doua linie a fișierul conține două numere naturale n și k cu semnificațiile din enunț.

Date de ieșire

• Dacă valoarea lui p este 1, se va rezolva numai punctul 1 din cerință. În acest caz, în fişierul de ieşire spirala2.out se va scrie un singur număr natural, reprezentând suma elementelor.
• Dacă valoarea lui p este 2, se va rezolva numai punctul 2 din cerință. În acest caz, în fişierul de ieşire spirala2.out se va scrie noua matrice cu proprietatea din cerință.

Restricții și precizări

• 2 ≤ n ≤ 500
• 1 ≤ k ≤ n
• Pentru rezolvarea corectă a primei cerinţe se acordă 30 de puncte, iar pentru cerința a doua se acordă 70 de puncte.

Exemplul 1:

spirala2.in

1
3 2

spirala2.out

20

Încărcare soluție

Lipește codul aici

import sys

sys.stdin = open('spirala2.intxt', 'r')
sys.stdout = open('spirala2.outtxt', 'w')

n, k, i, j, p, nr, v = 501, 501, 0, 0, 0, 0, [[0]*501 for _ in range(501)]
s, w, a, b, x, y = 0, [[0]*501 for _ in range(501)], 0, 0, 0, 0

p, n, k = map(int, input().split())

for a in range(1, n+1):
    x = a
    y = n - a + 1
    for i in range(x, y+1):
        nr += 1
        v[x][i] = nr
    for i in range(x+1, y+1):
        nr += 1
        v[i][y] = nr
    for i in range(y-1, x-1, -1):
        nr += 1
        v[y][i] = nr
    for i in range(y-1, x, -1):
        nr += 1
        v[i][x] = nr

if p == 1:
    for i in range(1, n+1):
        if i != k:
            s += v[i][k] % 100003 + v[k][i] % 100003
    print(s)
else:
    for i in range(1, n+1):
        w[i][k] = v[k][i]
        w[k][i] = v[i][k]
    for i in range(1, n+1):
        for j in range(1, n+1):
            if not w[i][j]:
                print(v[i][j], end=' ')
            else:
                print(w[i][j], end=' ')
        print()