2322 - Merry Cristmas

Sursa: [1]

Cerinţa

Într-un ținut îndepărtat cu peisaje de poveste pacea domnește peste întreg ținutul deoarece lumea are absolut tot ce îi trebuie.

Moș Crăciun însa dorește să le mai aducă zâmbetul pe buze și de aceea a decis să ducă fiecărei case câte o sticlă de șampanie deoarece deși oamenii de aici au bani, nu au magazine de unde să cumpere. Însă din păcate Moș Crăciun este bătrân și nu poate transporta prea multe sticle așa că decide să distribuie câte o sticla de șampanie la fiecare două case. Pentru simplitate Moș Crăciun a făcut o hartă a caselor în care o casă este notată cu 1 și pământul gol cu 0. Moș Crăciun însă nu prea le are cu programarea și de aceea vă roagă să îl ajutați!

Scrieți un program care primind harta caselor afișează câte sticle de șampanie trebuie să aducă Moș Crăciun.

Date de intrare

Programul va citi un număr natural n de pe prima linie și de pe a doua linie n numere naturale din intervalul [0;1].

Date de ieșire

Fișierul instabook.out va conține n linii: pe linia i se vor afișa separate printr-un spațiu cel mai slab loc si cel mai bun loc din clasament al pozei i, (1 ≤ i ≤ n), iar în consolă se va afișa mesajul "Datele introduse corespund cerintelor." În caz contrar programul va afisa "Datele introduse nu corespund cerintelor."

Restricţii şi precizări

• 1 ⩽ n ⩽ 100.000
• 1 ⩽ m ⩽ 200.000

Exemplul 1

Intrare

instabook.in

3 5

3

2

1

3

3

Ieșire

Datele introduse corespund cerintelor.

instabook.in

1 2

2 3

1 3

Exemplul 2

Intrare

instabook.in

0 0

0

0

0

0

0

Ieșire

Datele introduse nu corespund cerintelor.

Rezolvare

<syntaxhighlight lang="python" line>

1. 3196

NMAX = 100005

def validate_input(n, m):

   if not (1 <= n <= 100000):
       print("Datele introduse nu corespund cerintelor.")
       exit()
   if not (1 <= m <= 200000):
       print("Datele introduse nu corespund cerintelor.")
       exit()
   print("Datele introduse corespund cerintelor.")
   

def read_input(filename):

   with open(filename, "r") as fin:
       n, m = map(int, fin.readline().split())
       validate_input(n, m)
       a = [0] * NMAX
       p = [0] * NMAX
       ma = [0] * NMAX
       mi = [0] * NMAX
       for i in range(1, n + 1):
           a[i] = i
           p[i] = i
           ma[i] = i
           mi[i] = i
       for i in range(1, m + 1):
           x = int(fin.readline())
           if p[x] != 1:
               a1 = a[p[x] - 1]
               a2 = a[p[x]]
               a[p[x] - 1] = a2
               a[p[x]] = a1
               p[a1] += 1
               p[a2] -= 1
               ma[a1] = max(ma[a1], p[a1])
               mi[a2] = min(mi[a2], p[a2])
   return n, ma, mi

def write_output(filename, n, ma, mi):

   with open(filename, "w") as fout:
       for i in range(1, n + 1):
           fout.write(str(mi[i]) + ' ' + str(ma[i]) + '\n')

if __name__ == "__main__":

   n, ma, mi = read_input("instabook.in")
   write_output("instabook.out", n, ma, mi)

</syntaxhighlight>

Explicatie rezolvare

Acest program citeste date de intrare dintr-un fisier instabook.in si apoi le valideaza prin intermediul functiei validate_input, care verifica daca n si m respecta limitele impuse de cerinta. Daca datele de intrare sunt invalide, se afiseaza un mesaj corespunzator si programul se opreste prin intermediul functiei exit(). Daca datele sunt valide, se afiseaza un mesaj corespunzator si programul continua executia.

In continuare, programul aloca cate o lista pentru a, p, ma si mi de dimensiune NMAX. Lista a retine ordinea initiala a elementelor, lista p retine pozitia curenta a fiecarui element, lista ma retine pozitia maxima pe care a fost gasit fiecare element si lista mi retine pozitia minima pe care a fost gasit fiecare element.

Urmeaza citirea efectiva a elementelor si executia operatiilor de interschimbare. Daca elementul citit nu se afla deja pe pozitia 1 (adica nu este primul element), se efectueaza interschimbarea elementelor aflate pe pozitiile p[x]-1 si p[x]. Dupa aceasta, pozitia fiecarui element modificat este actualizata in lista p, iar pozitia maxima si minima pe care a fost gasit fiecare element sunt actualizate in listele ma si mi.

In final, programul scrie datele de iesire in fisierul instabook.out utilizand functia write_output. Aceasta parcurge toate elementele si scrie, pentru fiecare dintre ele, valorile minime si maxime ale pozitiei acestora in listele mi si ma.