2167 - alee

De la Universitas MediaWiki

Enunt

Parcul oraşului a fost neglijat mult timp, astfel că acum toate aleile sunt distruse. Prin urmare, anul acesta Primăria şi-a propus să facă reamenajări. Parcul are forma unui pătrat cu latura de n metri și este înconjurat de un gard care are exact două porți. Proiectanții de la Primărie au realizat o hartă a parcului și au trasat pe hartă un caroiaj care împarte parcul în nxn zone pătrate cu latura de 1 metru. Astfel harta parcului are aspectul unei matrice pătratice cu n linii și n coloane. Liniile și respectiv coloanele sunt numerotate de la 1 la n. Elementele matricei corespund zonelor pătrate de latură 1 metru. O astfel de zonă poate să conțină un copac sau este liberă. Edilii orașului doresc să paveze cu un număr minim de dale pătrate cu latura de 1 metru zonele libere (fără copaci) ale parcului, astfel încât să se obțină o alee continuă de la o poartă la alta.

Cerința

Scrieți un program care să determine numărul minim de dale necesare pentru construirea unei alei continue de la o poartă la cealaltă.

Date de intrare

Fișierul de intrare aleeIN.txt conține pe prima linie două valori naturale n și m separate printr-un spațiu, reprezentând dimensiunea parcului, respectiv numărul de copaci care se găsesc în parc. Fiecare dintre următoarele m linii conține câte două numere naturale x și y separate printr-un spațiu, reprezentând pozițiile copacilor în parc (x reprezintă linia, iar y reprezintă coloana zonei în care se află copacul). Ultima linie a fișierului conține patru numere naturale x1 y1 x2 y2, separate prin câte un spațiu, reprezentând pozițiile celor două porți (x1, y1 reprezintă linia și respectiv coloana zonei ce conține prima poartă, iar x2, y2 reprezintă linia și respectiv coloana zonei ce conține cea de a doua poartă).

Date de ieșire

Fișierul de ieșire aleeOUT.txt va conţine o singură linie pe care va fi scris un număr natural care reprezintă numărul minim de dale necesare pentru construirea aleii. În cazul în care restricțiile nu sunt îndeplinite, se va afișa mesajul "Datele nu corespund restrictiilor impuse".

Restricții și precizări

  • 1 ≤ n ≤ 175
  • 1 ≤ m < n*n
  • Aleea este continuă dacă oricare două plăci consecutive au o latură comună.
  • Aleea începe cu zona unde se găsește prima poartă și se termină cu zona unde se găsește cea de a doua poartă.
  • Pozițiile porților sunt distincte şi corespund unor zone libere.
  • Pentru datele de test există întotdeauna soluție.

Exemplul 1:

aleeIN.txt

8 6 
2 7
3 3
4 6
5 4
7 3
7 5 
1 1 8 8

aleeOUT.txt

15

Explicație

O modalitate de a construi aleea cu număr minim de dale este:

OOO-----

--OO--x-

--xO----

---OOx--

---xO---

----OO--

--x-xOO-

------OO

(cu X am marcat copacii, cu - zonele libere, iar cu O dalele aleii).

Exemplul 2:

aleeIN.txt

200 6 
2 7
3 3
4 6
5 4
7 3
7 5 
1 1 8 8

aleeOUT.txt

Datele nu corespund restrictiilor impuse

Rezolvare

from collections import deque

dx = [-1, 0, 0, 1]
dy = [0, -1, 1, 0]

def read():
    global n, m, x1, y1, x2, y2, a
    with open("aleeIN.txt", "r") as f:
        n, m = map(int, f.readline().split())
        if not check_restrictions():
            return False
        a = [[0] * (n + 1) for _ in range(n + 1)]
        for _ in range(m):
            i, j = map(int, f.readline().split())
            a[i][j] = -1
        x1, y1, x2, y2 = map(int, f.readline().split())
    return True

def check_restrictions():
    if not (1 <= n <= 175):
        with open("aleeOUT.txt", "w") as f:
            f.write("Datele nu corespund restrictiilor impuse\n")
        return False
    if not (1 <= m < n * n):
        with open("aleeOUT.txt", "w") as f:
            f.write("Datele nu corespund restrictiilor impuse\n")
        return False
    return True

def OK(i, j):
    if i < 1 or i > n or j < 1 or j > n:
        return False
    if a[i][j] == -1:
        return False
    return True

def lee():
    global a
    Q = deque([(x1, y1)])
    a[x1][y1] = 1
    while Q:
        i, j = Q.popleft()
        for k in range(4):
            ni = i + dx[k]
            nj = j + dy[k]
            if OK(ni, nj) and a[ni][nj] < 1:
                a[ni][nj] = a[i][j] + 1
                Q.append((ni, nj))

def main():
    if not read():
        return
    lee()
    with open("aleeOUT.txt", "w") as f:
        f.write(str(a[x2][y2]) + "\n")

if __name__ == "__main__":
    main()