Return sorted values after quadratic transform | Two Sigma Interview Question