ZA-SAMO 2010 Problem 1

For a positive integer $n$ , $S(n)$ denotes the sum of its digits and $U(n)$ its unit digit. Determine all positive integers $n$ with the property that $$n = S(n) + U(n)^2.$$