HackerRank Zeros and Ones in python problem solution - TheCScience