2183 - betisoare1

Enunț

Radu are o grămadă de bețișoare de două mărimi diferite. Cele cu mărime mai mică sunt marcate cu 0 și vom spune că sunt de tipul 0, iar celelalte sunt marcate cu 1 și vom spune că sunt de tipul 1. Grămada are N bețișoare, N număr natural. Radu se gândește să așeze pe un singur rând toate bețișoarele din grămadă, unul după altul, astfel încât bețișoarele formează secvențe de cifre 0 și 1. Apoi își propune să determine numărul total de secvențe care conțin un număr maxim de bețișoare de aceeași mărime.

Cerința

Scrieți un program care să citească numărul natural N și mărcile bețișoarelor, iar apoi să determine secvențele ce conțin un număr maxim de bețișoare de același tip.

Date de intrare

Fișierul de intrare betisoare1in.txt conține pe prima linie numărul natural N reprezentând numărul de bețișoare din grămadă, iar pe linia următoare cele N mărci ale bețișoarelor b[1], b[2],…, b[N] fără spații.

Date de ieșire

Fișierul de ieșire betisoare1out.txt va conține pe prima linie marca secvenței de bețișoare cu lungimea maximă, urmată de spațiu și apoi numărul maxim de bețișoare din secvența. Pe rândul următor se va afișa numărul de secvențe cu proprietatea cerută. Dacă există secvențe cu același număr maxim de bețișoare de tipul 0 și 1, se va afișa secvența de tipul 1.

Restricții și precizări

• 1 ⩽ N ⩽ 1000
• 0 ⩽ b[i] ⩽ 1, (1 ⩽ i ⩽ N)

Exemplul 1

Intrare

betisoare1in.txt

10

0100011000

betisoare1out.txt

Datele de intrare corespund restricțiilor impuse

0 3

2

Explicație

Secvența cea mai lungă cu bețișoare având aceeași marcă, este secvența formată cu bețișoare de tipul 0. Numărul maxim de bețișoare din secvență este 3. Există două asemenea secvențe.

Exemplul 2

Intrare

betisoare1in.txt

2

02

betisoare1out.txt

Marca fiecarui betisor trebuie sa fie 0 sau 1.

Rezolvare

#2183 - betisoare1
def validare_date_intrare(N, betisoare):
    if not (1 <= N <= 1000):
        print("Numarul de betisoare trebuie sa fie intre 1 si 1000.")
        return False

    if not all(0 <= b <= 1 for b in betisoare):
        print("Marca fiecarui betisor trebuie sa fie 0 sau 1.")
        return False

    return True


def secvente_maxime(N, betisoare):
    secventa_maxima = {"marca": -1, "lungime": 0}
    numar_secvente_maxime = 0

    marca_curenta = betisoare[0]
    lungime_curenta = 1

    for i in range(1, N):
        if betisoare[i] == marca_curenta:
            lungime_curenta += 1
        else:
            if lungime_curenta > secventa_maxima["lungime"]:
                secventa_maxima["lungime"] = lungime_curenta
                secventa_maxima["marca"] = marca_curenta
                numar_secvente_maxime = 1
            elif lungime_curenta == secventa_maxima["lungime"]:
                numar_secvente_maxime += 1

            marca_curenta = betisoare[i]
            lungime_curenta = 1

    if lungime_curenta > secventa_maxima["lungime"]:
        secventa_maxima["lungime"] = lungime_curenta
        secventa_maxima["marca"] = marca_curenta
        numar_secvente_maxime = 1
    elif lungime_curenta == secventa_maxima["lungime"]:
        numar_secvente_maxime += 1

    return secventa_maxima, numar_secvente_maxime


def main():
    with open("betisoare1in.txt", "r") as f:
        N = int(f.readline().strip())
        betisoare = list(map(int, f.readline().strip()))

    if not validare_date_intrare(N, betisoare):
        return
    if validare_date_intrare(N, betisoare):
        print("Datele de intrare corespund restricțiilor impuse")

    secventa_maxima, numar_secvente_maxime = secvente_maxime(N, betisoare)

    with open("betisoare1out.txt", "w") as g:
        g.write(f"{secventa_maxima['marca']} {secventa_maxima['lungime']}\n")
        g.write(str(numar_secvente_maxime))


if __name__ == "__main__":
    main()