Countable and uncountable sets

Let $A = \{0,1,2\}, B = \{5,6,9\}$.

The function $f(x) = x^2$ doesn't build a one-to-one correspondence between these two sets. How should you change it to build a one-to-one correspondence?

Write only the right part of your function $f(x) = ...$.