3sum Leetcode Solution Python