Non-zero (Codeforces 1300A)
Tags: Math, implementation

Guy-Manuel and Thomas have an array $$$a$$$ of $$$n$$$ integers [$$$a_1, a_2, \dots, a_n$$$]. In one step they can add $$$1$$$ to any element of the array. Formally, in one step they can choose any integer index $$$i$$$ ($$$1 \le i \le n$$$) and do $$$a_i := a_i + 1$$$.If either the sum or the product of all elements in the array is equal to zero, Guy-Manuel and Thomas do not mind to do this operation one more time.What is the minimum number of steps they need to do to make both the sum and the product of all elements in the array different from zero? Formally, find the minimum number of steps to make $$$a_1 + a_2 +$$$ $$$\dots$$$ $$$+ a_n \ne 0$$$ and $$$a_1 \cdot a_2 \cdot$$$ $$$\dots$$$ $$$\cdot a_n \ne 0$$$.

Input: Each test contains multiple test cases. The first line contains the number of test cases $$$t$$$ ($$$1 \le t \le 10^3$$$). The description of the test cases follows.The first line of each test case contains an integer $$$n$$$ ($$$1 \le n \le 100$$$) — the size of the array.The second line of each test case contains $$$n$$$ integers $$$a_1, a_2, \dots, a_n$$$ ($$$-100 \le a_i \le 100$$$) — elements of the array .

Output: For each test case, output the minimum number of steps required to make both sum and product of all elements in the array different from zero.

Note: In the first test case, the sum is $$$0$$$. If we add $$$1$$$ to the first element, the array will be $$$[3,-1,-1]$$$, the sum will be equal to $$$1$$$ and the product will be equal to $$$3$$$.In the second test case, both product and sum are $$$0$$$. If we add $$$1$$$ to the second and the third element, the array will be $$$[-1,1,1,1]$$$, the sum will be equal to $$$2$$$ and the product will be equal to $$$-1$$$. It can be shown that fewer steps can't be enough.In the third test case, both sum and product are non-zero, we don't need to do anything.In the fourth test case, after adding $$$1$$$ twice to the first element the array will be $$$[2,-2,1]$$$, the sum will be $$$1$$$ and the product will be $$$-4$$$.

Sample Inputs:
4
3
2 -1 -1
4
-1 0 0 1
2
-1 2
3
0 -2 1

Sample Outputs:
1
2
0
2

Solution (Python):
def main():
    for _ in range(int(input())):
        size = input()
        numbers = tuple(map(int, input().split()))
        zero = 0
        plus = 0
        minus = 0
        for i in numbers:
            if not i:
                zero += 1
            elif i > 0:
                plus += i
            else:
                minus += i
        plus += zero
        print(zero + (plus + minus == 0))
if __name__ == '__main__':
    main()

Solution (Python):
""""""

def i():
    n = int(input())
    s = list(map(int, input().split()))
    c = s.count(0)
    if sum(s, c) == 0:
        c += 1
    return str(c)
t = int(input())
a = []
for _ in range(t):
    a.append(i())
print('\n'.join(a))

Solution (Python):
def main():
    from sys import stdin
    e = stdin.readline
    for _ in range(int(e())):
        n = int(e())
        (a, b, r, sp) = ([], [], 0, 0)
        for x in map(int, e().split()):
            sp += 1 if x == 0 else x
            if x == 0:
                b.append(1)
                r += 1
            elif x > 0:
                b.append(x)
            else:
                a.append(x)
        if sp != 0:
            print(r)
        elif b:
            print(r + 1)
        else:
            a.sort()
            if a[0] < -1:
                print(r + 1)
            else:
                r += len(a) * 2
                print(r)
main()