Basic Array Program In Leetcode 1501