IMC 1997 · Problem 3

Let $A$ and $B$ be real $n \times n$ matrices such that $A^2 + B^2 = AB$. Prove that if $BA - AB$ is an invertible matrix then $n$ is divisible by $3$.