CF898E Squares and not squares

题目描述

Ann 和 Borya 拥有 $n$ 堆糖果,$n$ 是一个偶数。第 $i$ 堆有 $a_{i}$ 颗糖果。 Ann 喜欢数量为某个整数的完全平方数的糖果堆,而 Borya 不喜欢数量为任何整数的完全平方数的糖果堆。在每一步操作中,他们可以选择一堆糖果,并向其中增加一颗糖果(这颗糖果是新的,不属于任何其他堆),或者从其中移除一颗糖果(如果该堆至少有一颗糖果)。 请你求出,为了让恰好有 $n/2$ 堆糖果的数量是某个整数的完全平方数,且恰好有 $n/2$ 堆糖果的数量不是任何整数的完全平方数,最少需要多少步操作。如果一开始就满足条件,则输出 0。

输入格式

第一行包含一个偶数整数 $n$($2 \leq n \leq 200000$),表示糖果堆的数量。 第二行包含 $n$ 个整数 $a_{1}, a_{2}, ..., a_{n}$($0 \leq a_{i} \leq 10^9$),表示每堆糖果的数量。

输出格式

输出满足要求所需的最少操作次数。如果初始状态已经满足条件,则输出 0。

说明/提示

在第一个样例中,你可以通过两次操作达到条件。每次操作向第二堆糖果添加一颗糖果,最后第二堆的数量将变为 $16$。之后,第 $2$ 堆和第 $4$ 堆的糖果数是完全平方数,第 $1$ 堆和第 $3$ 堆的糖果数都不是完全平方数,因此条件得到满足。 在第二个样例中,你需要向任意三堆中各添加两颗糖果。 由 ChatGPT 5 翻译