3183 - RATC

Sursa: [1]

Cerinţa

Urmărind traseul autobuzului și cunoscând câte persoane urcă și coboară la fiecare stație, să se afle:

a) Numărul total de persoane care au urcat în autobuz, numărul total de persoane care au coborât din autobuz și numărul de persoane aflate în autobuz după terminarea traseului.

b) Știind că Alexandru se află la stația cu numărul de ordin x, să se afle dacă autobuzul în care urmează să urce este aglomerat. Un autobuz este considerat aglomerat dacă în el se află cel puțin y persoane înainte să ajungă în stație.

Date de intrare

Programul citește de la tastatură numerele naturale n x y p și apoi n perechi de numere Xi, Yi naturale care reprezintă numărul de persoane care au urcat, respectiv coborât la stația i.

Date de ieșire

Dacă p = 1, atunci se va rezolva numai cerința a). În acest caz, se vor scrie pe aceeași linie 3 numere separate prin câte un spațiu, cu semnificația din enunț.

Dacă p = 2, atunci se va rezolva numai cerința b). În acest caz, se va scrie pe prima linie DA dacă autobuzul este aglomerat, respectiv NU dacă autobuzul nu este aglomerat.

Restricţii şi precizări

1 ⩽ n ⩽ 1000
1 ⩽ x ⩽ n
1 ⩽ y ⩽ 10000
0 ⩽ Xi, Yi ⩽ 2000

Exemplul 1

Intrare: 5 2 10 1; 3 0; 2 1; 5 6; 4 3; 3 1
Ieșire: 17 11 6

Exemplul 2

Intrare: 5 2 10 2; 3 0; 2 1; 5 6; 4 3; 3 1
Ieșire: NU

Rezolvare

3183

def read_input(n):

   """Citeste valorile de x si y pentru n obiecte si calculeaza suma x si suma y"""
   u, c = 0, 0
   for i in range(n):
       x, y = map(int, input().split())
       u += x
       c += y
   return u, c

def validate_input(n, x, y):

   """Validează input-ul în funcție de condițiile date"""
   if not (1 <= n <= 1000):
       print("Eroare: n trebuie să fie între 1 și 1000")
       return False
   if not (1 <= x <= n):
       print("Eroare: x trebuie să fie între 1 și n")
       return False
   if not (1 <= y <= 10000):
       print("Eroare: y trebuie să fie între 1 și 10000")
       return False
   return True

def calculate_output_1(n, x, y, p):

   """Calculeaza output-ul pentru cazul in care p=1"""
   u, c = read_input(n)
   print(u, c, u-c)

def calculate_output_2(n, x, y, p):

   """Calculeaza output-ul pentru cazul in care p=2"""
   u = 0
   for i in range(1, x):
       a, b = map(int, input().split())
       u += a - b
   for i in range(x, n+1):
       a, b = map(int, input().split())
   if u < y:
       print("NU")
   else:
       print("DA")

def main():

   n, x, y, p = map(int, input().split())
   if not validate_input(n, x, y):
       return
   if p == 1:
       calculate_output_1(n, x, y, p)
   else:
       calculate_output_2(n, x, y, p)

main()

</syntaxhighlight>