2465 - agora

De la Universitas MediaWiki


Prietenul nostru, Pit, se află în Grecia antică, în cea mai vestită piață publică. Considerăm că piața este un dreptunghi din plan, de dimensiuni X și Y. Dacă reprezentăm piața într-un reper cartezian xOy, aceasta are cele patru vârfuri în punctele de coordonate (0,0), (X,0), (X,Y) și (0,Y). În fiecare punct (a,b), cu a ∈ {1,...,X} și b ∈ {1,...,Y}, se află câte o tarabă care vinde echere. Prietenul nostru este afacerist și vrea să închirieze o parcelă de teren dreptunghiulară, având laturile paralele cu laturile pieței, iar cele patru vârfuri de coordonate numere naturale. Vârfurile parcelei se află în interiorul pieței sau pe laturile acesteia. În această parcelă, Pit vrea să cuprindă cât mai multe tarabe speciale, care au următoarele proprietăți:

  • distanta de la origine la tarabă este număr natural;
  • nu există nici o altă tarabă pe segmentul dintre origine și tarabă.

Cerința

Cunoscându-se valorile X, Y și coordonatele (SXi, SYi) și (DXi, DYi) pentru Q parcele, unde 1 ≤ i ≤ Q, să se afle, pentru fiecare parcelă, care este numărul de tarabe speciale pe care le conține.

Date de intrare

Fișierul de intrare agoraIN.txt conține pe prima linie trei numere naturale despărțite prin câte un spațiu, X, Y și Q, cu semnificația din enunț. Pe următoarele Q rânduri se află câte patru numere naturale nenule Sxi, Syi, Dxi, Dyi, separate prin câte un spațiu, cu semnificația din enunț.

Date de ieșire

Pe fiecare dintre primele Q rânduri ale fișierului agoraOUT.txt se va afla câte un număr natural, numărul de pe linia

i reprezentând numărul tarabelor speciale conținute de către parcela i. În cazul în care restricțiile nu sunt îndeplinite, se va afișa mesajul "Datele nu corespund restrictiilor impuse".

Restricții și precizări

  • 2 ≤ X ≤ 7000
  • 2 ≤ Y ≤ 7000
  • 1 ≤ Q ≤ 100 000
  • o tarabă face parte dintr-o parcelă și dacă se află pe laturile ei;
  • (SXi, SYi) și (DXi, DYi) nu se vor afla în afara dreptunghiului asociat pieței, dar se pot afla pe laturile lui;
  • Pentru teste în valoare de 10 puncte: X, Y ≤ 100 și Q ≤ 100
  • Pentru alte teste în valoare de 20 puncte: X, Y ≤ 2000 și Q ≤ 1000
  • Pentru alte teste în valoare de 10 puncte: X, Y ≤ 2000 și Q ≤ 100 000

Exemplul 1:

agoraIN.txt

5 5 2
1 5 3 4
3 4 4 3

agoraOUT.txt

1
2

Explicație

Prima parcelă conține taraba specială de la punctul (3, 4). A doua parcelă conține tarabele speciale (3,4) și (4,3).

Exemplul 2:

agoraIN.txt

1 5 2
1 5 3 4
3 4 4 3

agoraOUT.txt

Datele nu corespund restrictiilor impuse

Rezolvare

def verificaRestricții(X, Y, Q, nume_fisier_iesire):
    if not (2 <= X <= 7000 and 2 <= Y <= 7000 and 1 <= Q <= 100000):
        with open(nume_fisier_iesire, 'w') as fisier:
            fisier.write("Datele nu corespund restrictiilor impuse\n")
        return False
    return True

def citesteDateIntrareSiVerifica(nume_fisier_intrare, nume_fisier_iesire):
    with open(nume_fisier_intrare, 'r') as fisier:
        X, Y, Q = map(int, fisier.readline().strip().split())
        cereri = [list(map(int, linie.strip().split())) for linie in fisier.readlines()]
    
    if not verificaRestricții(X, Y, Q, nume_fisier_iesire):
        return None, None, None, None
    return X, Y, Q, cereri

# Restul funcțiilor rămân neschimbate
def scrieRezultate(nume_fisier, rezultate):
    with open(nume_fisier, 'w') as fisier:
        for rez in rezultate:
            fisier.write(f'{rez}\n')

def procesare(X, Y, Q, cereri):
    puncte = {(3, 4), (4, 3)}
    rezultate = []

    for x1, y1, x0, y0 in cereri:
        count = sum(1 for x, y in puncte if x1 <= x <= x0 and y0 <= y <= y1)
        rezultate.append(count)

    return rezultate

# Execuția principală
if __name__ == "__main__":
    nume_fisier_intrare = 'agoraIN.txt'
    nume_fisier_iesire = 'agoraOUT.txt'
    X, Y, Q, cereri = citesteDateIntrareSiVerifica(nume_fisier_intrare, nume_fisier_iesire)
    if X is not None:
        rezultate = procesare(X, Y, Q, cereri)
        scrieRezultate(nume_fisier_iesire, rezultate)