Editing 1865 - Summit

== Cerința ==
Se dă un şir <code>x</code> format din <code>n</code> numere naturale nenule. Pentru fiecare element <code>xi</code> din şir să se verifice dacă există un număr <code>k</code> astfel încât elementul <code>xi</code> să fie egal cu suma primelor <code>k</code> elemente din şir.

== Date de intrare ==
Fișierul de intrare <code>summitIN.txt</code> conține pe prima linie numărul <code>n</code>, iar pe a doua linie <code>n</code> numere naturale separate prin spații.

== Date de ieșire ==
Fișierul de ieșire <code>summitOUT.txt</code> va conține pe linia <code>i</code> valoarea <code>k</code> dacă elementul <code>xi</code> este egal cu suma primelor <code>k</code> elemente din şir, sau <code>0</code> în caz contrar, pentru fiecare <code>i</code> de la <code>1</code> la <code>n</code>.

== Restricții și precizări ==

* <code>2 ≤ n ≤ 1.000.000</code>
* numerele de pe a doua linie a fișierului de intrare vor fi mai mici decât <code>2.000.000.000</code>

== Exemplul 1 ==
<code>summitIN.txt</code>
 3
 1 2 3
<code>summitOUT.txt</code>
 1
 0
 2

== Exemplul 2 ==
<code>summitIN.txt</code>
 1
 1
Numerele introduse nu respectă restricțiile.

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def verifica_restrictii(n, nums):
    return 2 <= n <= 1000000 and all(0 < x < 2000000000 for x in nums)

def verifica_summit(n, nums):
    suma_partiala = [0]  # Lista pentru suma parțială, începând cu 0

    for i in range(n):
        suma_partiala.append(suma_partiala[-1] + nums[i])

    rezultate = []
    for i in range(n):
        k = 0
        while k < n and suma_partiala[k] < nums[i]:
            k += 1

        if k < n and suma_partiala[k] == nums[i]:
            rezultate.append(k)
        else:
            rezultate.append(0)

    return rezultate

def main():
    with open("summitIN.txt", "r") as f:
        n = int(f.readline())
        nums = list(map(int, f.readline().split()))

    if verifica_restrictii(n, nums):
        rezultate = verifica_summit(n, nums)

        with open("summitOUT.txt", "w") as f:
            for rez in rezultate:
                f.write(str(rez) + "\n")
    else:
        print("Numerele introduse nu respectă restricțiile.")

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