equal_stacks.py

# Source - https://www.hackerrank.com/challenges/equal-stacks/problem

def allSame(arr):
    return arr[0]["total"] == arr[1]["total"] == arr[2]["total"]

def equalStacks(h1, h2, h3):
    if(len(h1) == 0 or len(h2) == 0 or len(h3) == 0):
        return 0;
    stacks = [
        {"arr": h1, "total":sum(h1)},
        {"arr": h2, "total":sum(h2)},
        {"arr": h3, "total":sum(h3)},
    ]

    if(allSame(stacks)):
        return stacks[0]["total"]

    while(True):
        highest = stacks[0]
        for stack in stacks:
            if(stack["total"] > highest["total"]):
                highest = stack

        highest["total"] -= highest["arr"].pop(0)
        if(allSame(stacks)):
            return stacks[0]["total"]


    return stacks[0]["total"]

Hi, I would remove the validation at lines 15 and 25 and place it into the while condition like : while (not allSame(stacks)) { ... }