Python Leetcode 1004 Tv