- At one point, it says, Thus it is enough to assume $m\in A$. Is that correct? I think they meant $f(m)$ instead of $m$.
- Secondly, there's another error where they claim $\lfloor f(m) \rfloor =k$. It's supposed to be $\left\lfloor \sqrt f(m) \right\rfloor =k$
DOUBTS :
- In the very last line they say This clearly implies that $f^{2j}= (k+j)^2$ but I don't find it quite trivial. Can "you" please explain this part?
I think that the floored part is $k$, then, $k+1$, then $k+2$, $\cdots$. But then $f^{2j}$ doesn't come similar to theirs.

They do mean the $f(m)\in A$, and this is unclear because of the poor choice of notation because in the definition of A and B, the author uses $m$ as an element of the sets, but $m$ had already been used as the starting number.
I believe you are right about your second error.
You can notice that the remainder of $f^2(m)$ that is not a square is $j-1$ which is one less than $j$, and $k$ increased by 1 to $k+1$. The idea is that every time 2 compositions of $f$ is applied, the remainder will decrease by 1 and the square will increase by 1, so $f^{2j}(m) = (k+j)^2 + 0$.