https://remp3indir.net/hackerrank-equalize-the-array-python-solution/