
nVoters = 5

class Voter:
    input = None
    output = None

# Setup
voters = []
for i in range(nVoters):
    v = Voter()
    if rt.id == i+1:
        v.input = int(raw_input("Your vote (0/1): "))
    v.input = rt.shamir_share([i+1], v.input)
    voters.append(v)

def main(voters):
    vs = [0, 0]
    win = 0

    for i in range(0, nVoters):
        vc[0] = (vc[0] + 1) * voters[i].input + vc[0] * (1 - voters[i].input)
        vc[1] = (vc[1] + 1) * (1 - voters[i].input) + vc[1] * voters[i].input

    win = vc[1] > vc[0]
    for i in range(0, nVoters):
        voters[i].output = rt.open(win, [i+1])

