Let $V$ be a finite-dimensional vector space over $\mathbb{C}$ and let $f:V\rightarrow V$ be an invertible linear transformation. Prove that $f$ is diagonalizable if and only if $f^2$ (meaning $f\circ f$) is diagonalizable.
I tried considering the characteristic polynomial of the transformation since $f$ will be triagonalizable (as every endomorphism over $C$) meaning it's eigenvalues are just the elements of the diagonal, but I'm not sure if that helps...

Hint: if $AfA^{-1}$ is diagonal, then $Af^2A^{-1}=AfA^{-1}AfA^{-1}$ is a product of two diagonal elements, and therefore also diagonal.
The other direction is similar, but you have to use the fact that $f$ is invertible.