Editing 1701 - Birouri

Arhi şi-a propus să extindă clădirea de birouri pe care a proiectat-o iniţial pe un singur nivel numerotat cu <code>1</code>, împărţit în <code>n*n</code> zone pătratice de latură <code>1</code>, fiecare corespunzând unui birou, prin construirea mai multor niveluri. În colţurile tuturor birourilor se construiesc grinzi de rezistenţă. Pentru a asigura rezistenţa întregii clădiri, Arhi va proiecta niveluri noi, numerotate cu <code>2</code>, <code>3</code>,…  atât timp cât conțin cel puțin un birou și sunt respectate următoarele patru reguli:

* '''R1''': fiecare nivel nou va fi proiectat sub forma unui dreptunghi sau pătrat de arie maximă pentru nivelele cu număr impar, respectiv, sub forma unui pătrat de arie maximă pentru nivelele cu număr par;
* '''R2''': fiecare dintre colţurile zidurilor unui nivel nou trebuie plasat pe câte o grindă de rezistenţă dintre două sau mai multe birouri de pe nivelul precedent;
* '''R3''': oricare două dintre colţurile zidurilor unui nivel nou vor fi plasate pe ziduri diferite (un zid nu se poate suprapune în totalitate pe alt zid) şi cel puţin două vârfuri opuse ale unui nivel nou se vor afla pe ziduri opuse ale nivelului precedent;
* '''R4''': orice porţiune de zid de pe nivelul <code>k</code> (<code>k>1</code>), construită deasupra unui birou de pe nivelul <code>k-1</code>, se va suprapune exact peste una dintre laturile biroului, sau îl va străbate în diagonală.

Birourile de pe nivelul <code>k</code> (<code>k>1</code>), vor fi construite exact deasupra celor de pe nivelul precedent, astfel, nivelurile <code>2</code>, <code>4</code> etc. vor avea lângă ziduri spaţii triunghiulare care nu vor aparţine niciunui birou.

Numerele inscripţionate pe birouri în imaginea de mai sus, indică nivelul corespunzător birourilor vizibile de deasupra clădirii.

= Cerința =
Cunoscându-se lungimea <code>n</code> a laturii primului nivel al clădirii, să se determine:

# numărul maxim de niveluri pe care le poate avea clădirea;
# numărul total de birouri ale clădirii cu număr maxim de niveluri.

= Date de intrare =
Fișierul de intrare <code>birouri.in</code> conține pe prima linie una dintre valorile <code>1</code> sau <code>2</code>, reprezentând cerinţa 1, dacă se cere determinarea numărului maxim de niveluri pe care le poate avea clădirea, respectiv cerinţa 2, dacă se cere determinarea numărului total de birouri al clădirii cu număr maxim de niveluri.

Linia a doua conţine un număr natural <code>n</code> (reprezentând lungimea fiecărui zid al primului nivel al clădirii)

= Date de ieșire =
Fișierul de ieșire <code>birouri.out</code> va conține pe prima linie pe prima linie un număr natural reprezentând numărul maxim de niveluri pe care le poate avea clădirea, dacă cerinţa a fost <code>1</code>, respectiv un număr natural reprezentând numărul total de birouri ale clădirii cu număr maxim de niveluri, dacă cerinţa a fost <code>2</code>.

= Restricții și precizări =

* <code>3 ≤ n ≤ 32768</code>
* Pentru rezolvarea corectă a cerinţei 1 se acordă 30% din punctaj, iar pentru rezolvarea corectă a cerinţei 2 se acordă 70% din punctaj.

=== Explicație ===
Exemplul corespunde imaginii de mai sus. Clădirea cu nivelul de la bază de latură <code>10</code> va avea <code>5</code> niveluri.

Nivelul <code>6</code> nu se mai construieşte, deoarece nu ar conţine niciun birou.

=== Explicație ===
Clădirea cu <code>5</code> niveluri şi latura de la bază de lungime <code>10</code> are:

* pe primul nivel <code>100</code> birouri;
* pe nivelul doi <code>40</code> birouri;
* pe nivelul trei <code>24</code> birouri;
* pe nivelul patru <code>4</code> birouri;
* pe nivelul cinci <code>4</code> birouri.

<code>100 + 40 + 24 + 4 + 4 = 172</code>

== Încărcare soluție ==

=== Lipește codul aici ===
 n, m, k, c, niv, b, p = 0, 0, 0, 0, 0, 0, 0

 fin = open('birouri.in', 'r')

 fout = open('birouri.out', 'w')

 c, n = map(int, fin.readline().split())

 m = n

 while n != 0 and n % 2 == 0:

     niv += 1

     if niv % 2 == 1:

         b += n * m

     else:

         if n > m:

             k = m

         else:

             k = n

         p = k * k // 2 - k

         b += p

         if p == 0:

             niv -= 1

         if k % 4 == 0:

             k = k // 2

             n = k

             m = k

         else:

             n = n // 2 - 1

             m = n + 2

 if c == 1:

     print(niv, file=fout)

 else:

     print(b, file=fout)

 fin.close()

 fout.close()