1131 - Arc

Irinuca a descoperit un nou joc pe calculator. Pe ecran sunt plasate pe o linie n bile colorate. Culorile bilelor sunt codificate cu numere naturale. Un subșir de bile alăturate având toate aceeași culoare se numește secvență. O secvență va conține numărul maxim de bile alăturate având aceeași culoare. Lungimea unei secvențe este egală cu numărul de bile din care este compusă.

Irinuca are la dispoziție un arc special. Trăgând cu arcul asupra unei bile, dacă aceasta face parte dintr-o secvență de lungime cel puțin egală cu 3, întreaga secvență va fi eliminată, iar bilele din dreapta secvenței se vor deplasa spre stânga pentru a umple “golul” lăsat de bilele eliminate. Dacă imediat în stânga și în dreapta secvenței eliminate se găseau două secvențe având aceeași culoare și dacă secvența obținută din unirea acestora după eliminare are o lungime cel puțin egală cu 3, atunci va fi și ea eliminată la rândul ei. Procesul continuă până când secvențele din stânga și dreapta unei secvențe tocmai eliminate au culori diferite sau până când lungimea secvenței obținute prin alăturare este mai mică decât 3 sau până când în stânga ori în dreapta unei secvențe eliminate nu se mai găsesc bile sau până sunt eliminate toate bilele de pe ecran.

Scopul jocului este de a elimina cât mai multe bile de pe ecran. Cum Irinuca încă nu se pricepe prea bine la acest joc și-a stabilit o strategie. Va trage cu arcul întotdeauna asupra unei bile ce face parte din secvența de lungime maximă de pe ecran. Dacă sunt mai multe astfel de secvențe, ea va alege cea mai din stânga secvență de lungime maximă. Dacă toate secvențele de pe ecran au lungimi mai mici decât 3, Irinuca nu va mai putea elimina nici una din ele și jocul se încheie.

De exemplu, dacă șirul inițial de bile este

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

Irinuca va acționa asupra unei bile de culoare 2. Prin eliminare se obține șirul de bile

5 1 3 3 3 1 1 5 6 4 4 4 4 7

din care se elimină și secvența de bile de culoare 3 obținându-se șirul de bile

5 1 1 1 5 6 4 4 4 4 7

din care se elimină și secvența de culoare 1.

5 5 6 4 4 4 4 7

Cum secvența de bile de culoare 5 nu este suficient de lungă, aceasta nu se mai elimină. Acum Irinuca trage asupra unei bile de culoare 4 și obține

5 5 6 7

dar cum în stânga și în dreapta secvenței eliminate sunt secvențe de culori diferite, nu se va mai elimina nici o secvență. Jocul se încheie deoarece nu mai există nici o secvență de lungime cel puțin 3 asupra căreia să se poată trage.

Cerinţă

Cunoscând numărul de bile și culorile fiecărei bile de pe ecran se cere să se determine:

1. numărul de secvențe de bile care se aflau inițial pe ecran;

2. numărul de bile care rămân neeliminate de pe ecran și culorile bilelor rămase în ordine pe ecran la finalul jocului.

Date de intrare

Fișierul de intrare arc.in conține pe prima linie un număr natural V. Pentru toate testele de intrare, numărul V poate avea doar valoarea 1 sau 2.

A doua linie conține un număr natural n reprezentând numărul de bile, iar a treia linie conține n numere naturale c1, c2, …, cn separate prin câte un spațiu, reprezentând culorile celor n bile de pe ecran.

Date de ieșire

Dacă valoarea lui V este 1, se va rezolva numai punctul 1 din cerință.

În acest caz, în fişierul de ieşire arc.out se va scrie un singur număr natural n1, reprezentând numărul de secvențe de bile aflate inițial pe ecran.

Dacă valoarea lui V este 2, se va rezolva numai punctul 2 din cerință.

În acest caz, în fişierul de ieşire arc.out se va scrie pe prima linie un singur număr natural n2, reprezentând numărul de bile care rămân neeliminate de pe ecran la finalul jocului, iar pe următoarele n2 linii se va scrie câte un număr natural reprezentând în ordine culorile bilelor rămase neeliminate la finalul jocului.

Dacă la finalul jocului nu mai rămâne nici o bilă neeliminată, fișierul de ieșire va conține pe prima sa linie valoarea 0.

Restricții și precizări

• 1 ≤ n ≤ 10 000
• 1 ≤ c1, c2, …, cn ≤ 100 000
• Pentru rezolvarea corectă a primei cerinţe se acordă 20 de puncte, iar pentru cerința a doua se acordă 80 de puncte.

Exemplul 1

arc.in

1
18
5 1 3 3 2 2 2 2 3 1 1 5 6 4 4 4 4 7

arc.out

10

Explicație

V = 1

Atenție! Pentru acest test se rezolvă doar cerința 1.

Secvențele sunt (5), (1), (3, 3), (2,2,2,2), (3), (1,1), (5), (6), (4,4,4,4), (7).

Încărcare soluție

Lipește codul aici

<syntaxhighlight lang="python" line="1"> f = open("arc.in", "r") g = open("arc.out", "w")

p = 0 c = [0] * 10000 x = [0] * 100000 m = 0 n = 0

def citire():

   global p, n, c, x, m
   p = int(f.readline())
   n = int(f.readline())
   c[1] = int(f.readline())
   x[1] = 1
   m = 1
   for i in range(2, n+1):
       k = int(f.readline())
       if k == c[m]:
           x[m] += 1
       else:
           m += 1
           c[m] = k
           x[m] = 1

def maxim():

   mx = x[1]
   k = 1
   for i in range(1, m+1):
       if x[i] > mx:
           mx = x[i]
           k = i
   return k

def elim(k):

   global n, m
   if x[k] > 2:
       n -= x[k]
       i = k - 1
       j = k + 1
       while c[i] == c[j] and i >= 1 and j <= m and x[i] + x[j] > 2:
           n -= x[i] + x[j]
           i -= 1
           j += 1
       if j > m:
           m = i
       else:
           if i >= 1 and j <= m and c[i] == c[j]:
               x[i] += x[j]
               j += 1
           m1 = m - j + i + 1
           i += 1
           while j <= m:
               x[i] = x[j]
               c[i] = c[j]
               i += 1
               j += 1
           m = m1

citire() if p == 1:

   g.write(str(m) + "\n")

else:

   k = maxim()
   while n > 0 and x[k] > 2:
       elim(k)
       if k > 0:
           k = maxim()
   if n > 0:
       g.write(str(n) + " ")
       for i in range(1, m+1):
           for j in range(1, x[i]+1):
               g.write(str(c[i]) + " ")
   else:
       g.write("0 ")
   

f.close() g.close() </syntaxhighlight>