Edit Distance Leetcode Solution Python