Update solution to Three Sum (Leetcode)