Editing 3641 - Xorstanta (section)

= Exemplu: =
Intrare
 4
 6
 30
 15
 18
Ieșire
 7
 31
 30
 43
<syntaxhighlight lang="python3">
def calculeaza_suma_maxima_xor(n):
    # Setăm suma_max la 29, valoare constantă folosită în bucla for.
    suma_max = 29
    numar_a, numar_b = 0, 0

    # Calculăm valoarea lui a în funcție de restul împărțirii lui n la 4.
    if n % 4 == 1:
        numar_a = 1
    elif n % 4 == 2:
        numar_a = n + 1
    elif n % 4 == 3:
        numar_a = 0
    else:
        numar_a = n

    # Calculăm valoarea numarului b în funcție de puterile lui 2 până la 2^29 care sunt mai mici sau egale cu n.
    for j in range(suma_max, -1, -1):
        if (1 << j) <= n:
            numar_b += (1 << j)

    # Efectuăm operația XOR între a și b și returnăm suma rezultată.
    numar_a = numar_a ^ numar_b
    return numar_a + numar_b


def main():
    # Citim numărul de teste.
    numarul_t = int(input("T: "))

    # Parcurgem fiecare test.
    for _ in range(numarul_t):
        # Citim valoarea n pentru testul curent.
        numarul_n = int(input("n: "))

        # Calculăm și afișăm suma maximă XOR pentru valoarea n dată.
        suma_maxima_xor = calculeaza_suma_maxima_xor(numarul_n)
        print(suma_maxima_xor)


if __name__ == "__main__":
    main()
</syntaxhighlight>