The solution is correct, but I agree that the notation could be improved. I would write:
Suppose . Then . Taking the square root of both sides, we find . If , then is not in the domain, so . If , then .
The difference is implies either or , both of which mean . But, that is not actually what is meant. The notation I used is which is more general, as now it means or .