rewrite leetcode 3SUM closest