Equal Stacks

  • + 0 comments

    def equalStacks(h1, h2, h3):

    sums = [sum(h1), sum(h2), sum(h3)]
    stacks = [h1, h2, h3]
    
    t = min(sums)
    
    while sums[0] != sums[1] or sums[1] != sums[2] or sums[0] != sums[2]: 
    
        for i in range(len(sums)):
            while sums[i] > t:
                sums[i] -= stacks[i].pop(0)   
    
        t = min(sums)
    
    return t