Problem

Source: 2021 USAJMO Problem 1

Tags: function, USAJMO, functional equation



Let $\mathbb{N}$ denote the set of positive integers. Find all functions $f : \mathbb{N} \rightarrow \mathbb{N}$ such that for positive integers $a$ and $b,$ \[f(a^2 + b^2) = f(a)f(b) \text{ and } f(a^2) = f(a)^2.\]