Number Following A Pattern Leetcode Problems On Python