Python Subarray Leetcode Problem