https://codeworld19.blogspot.com/2020/10/arrays-in-python-hacker-rank-solution.html