How many birds in this population would be expected to have either red or white feathers?
The solution

$$p^2 = 202 / 630 ≈ 0.32$$
$$p = \sqrt{202/630}$$
$$(1-p) = 1 - \sqrt{202/630}$$
$$(1-p)^2 = \left(1 - \sqrt{202/630}\right)^2 ≈ 0.188$$

There is a fraction of $202 / 630 ≈ 0.32$ of red birds. There is a fraction of $\left(1 - \sqrt{202/630}\right)^2 ≈ 0.188$ of white birds. There is therefore a fraction of $0.32 + 0.188 = 0.508$ of birds that are either all white or all red. In a population of 630 individuals, it therefore represents $630 * 0.508 ≈ 320.53$. A result that is apparently rounded up at $321$ in your answer key.

Your mistake

You computed the number of birds that have both white and red feathers (WR) while the question was asking for either red or white feathers. You calculated the expected number of heterozygotes instead of calculating the expected number of homozygotes.

Of course the number of heterozygotes and the number of homozygotes should add up to the number of individuals in the population ($321 + 309 = 630$).