import random
def Distribute(ones, zeros):
leng = ones + zeros
err = leng // 2
res = []
for i in range(0, leng):
err = err - ones
if err < 0 :
res.append(1)
err = err + leng
else:
res.append(0)
return res
for i in range(100):
z = 1 + random.randint(1,120)
o = 1 + random.randint(1,120)
l = Distribute(o,z)
oo = sum(l)
zz = len(l) - oo
if (zz!=z) or (oo!=o):
print('failed at', o, z)# your code goes here
aW1wb3J0IHJhbmRvbQpkZWYgRGlzdHJpYnV0ZShvbmVzLCB6ZXJvcyk6CiAgICBsZW5nID0gb25lcyArIHplcm9zCiAgICBlcnIgPSBsZW5nIC8vIDIKICAgIHJlcyA9IFtdCiAgICBmb3IgaSBpbiByYW5nZSgwLCBsZW5nKToKICAgICAgICBlcnIgPSBlcnIgLSBvbmVzCiAgICAgICAgaWYgZXJyIDwgMCA6CiAgICAgICAgICAgIHJlcy5hcHBlbmQoMSkKICAgICAgICAgICAgZXJyID0gZXJyICsgbGVuZwogICAgICAgIGVsc2U6CiAgICAgICAgICAgIHJlcy5hcHBlbmQoMCkKICAgIHJldHVybiByZXMKCmZvciBpIGluIHJhbmdlKDEwMCk6CiAgICB6ID0gMSArIHJhbmRvbS5yYW5kaW50KDEsMTIwKQogICAgbyA9IDEgKyByYW5kb20ucmFuZGludCgxLDEyMCkKICAgIGwgPSBEaXN0cmlidXRlKG8seikKICAgIG9vID0gc3VtKGwpCiAgICB6eiA9IGxlbihsKSAtIG9vCiAgICBpZiAoenohPXopIG9yIChvbyE9byk6CiAgICAgICAgcHJpbnQoJ2ZhaWxlZCBhdCcsIG8sIHopIyB5b3VyIGNvZGUgZ29lcyBoZXJl