3499 - Secv011

Sursa: - Secv011

---

Cerinţa

Se dă un șir A format din n elemente din mulțimea {0,1}. Numim secvență a șirului A orice succesiune de elemente situate pe poziții consecutive în șir. Numim secvență a șirului A orice succesiune de elemente situate pe poziții consecutive în șir. Să se determine numărul de secvențe ale șirului A ce conțin de două ori mai mulți de 1 decât de 0.

Date de intrare

Pe prima linie a fișierului secv011.in se află numărul n. Pe următoarea linie se află cele n elemente ale șirului , separate prin câte un spațiu.

Date de ieșire

Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt corecte.", iar apoi in fișierul de ieșire secv011.out va conține pe prima linie numărul cerut. În caz contrar, se va afișa pe ecran: "Datele nu sunt comform restricțiilor impuse.".

Restricţii şi precizări

• 1 ≤ n ≤ 1.000.000

Exemple

Exemplul 1

secv011.in

6

1 0 1 1 1 0

Ieșire

Datele sunt corecte.

secv011.out

4

Exemplul 2

secv011.in

10

1 0 0 0 1 1 1 0 0 0

Ieșire

Datele sunt corecte.

secv011.out

2

Exemplul 3

secv011.in

2

314441 41241241

Ieșire

Datele nu sunt comform restricțiilor impuse.

Rezolvare

<syntaxhighlight lang="python" line>

1. 0302 Secv011

def secv011(vector,n):

   f = open("secv011.out","w")
   contor = 0 
   zero = 0 
   unu = 0 
   for i in range(n):
       for j in range(i,n):
           if vector[j] == 0:
               zero += 1
           else:
               unu +=1
           if unu > 1 and unu == 2 * zero:
               contor += 1
       zero = 0
       unu = 0
   f.write(str(contor))
   
   

def conform_restrictiilor():

   with open('secv011.in') as f:
       n = int(f.readline().strip())
       vector = list(map(int, f.read().split()))
   if not 1 <= n <= 100000:
       print("Datele nu sunt conform restricțiilor impuse.")
       exit()
   for x in vector:
       if x > 1:
           print("Datele nu sunt conform restricțiilor impuse.")
           exit()
   print("Datele sunt corecte.")
   return n, vector 

if __name__ == '__main__':

   n, vector= conform_restrictiilor()
   secv011(vector,n)

</syntaxhighlight>