Python Subarray Leetcode Problem Set 0