Leetcode Two Sum Problem Python Snake