MathCo interview question

What would I do differently?