Editing 0874 - Atomi (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
def numara_atomi(expresie):
    stack = []
    i = 0
    n = len(expresie)
    
    while i < n:
        if expresie[i] == '(':
            stack.append('(')
            i += 1
        elif expresie[i] == ')':
            sub_exp = []
            while stack and stack[-1] != '(':
                sub_exp.append(stack.pop())
            stack.pop()  # scoatem '('
            sub_exp.reverse()
            i += 1
            coef = 0
            while i < n and expresie[i].isdigit():
                coef = coef * 10 + int(expresie[i])
                i += 1
            coef = max(coef, 1)
            for elem in sub_exp:
                stack.append((elem[0], elem[1] * coef, elem[2] * coef))
        elif expresie[i] in 'AB':
            tip = expresie[i]
            i += 1
            numar = 0
            while i < n and expresie[i].isdigit():
                numar = numar * 10 + int(expresie[i])
                i += 1
            numar = max(numar, 1)
            if tip == 'A':
                stack.append(('A', numar, 0))
            else:
                stack.append(('B', 0, numar))
        else:
            i += 1
    
    total_A = 0
    total_B = 0
    for elem in stack:
        total_A += elem[1]
        total_B += elem[2]
    
    return total_A, total_B

def main():
    expresie = input().strip()
    
    # Verificăm restricția privind lungimea expresiei
    assert len(expresie) <= 255, "Expresia trebuie să aibă cel mult 255 de caractere"
    
    nA, nB = numara_atomi(expresie)
    print(f"{nA} {nB}")

if __name__ == "__main__":
    main()


</syntaxhighlight>