每类人的贡献是独立的。假设我们知道总共分了 x 组,那么我们可以枚举每类人计算其贡献。将 S 个同类人分在 x 组内的贡献是组内人数两两乘积和的 b 倍,平方公式拆一下可知我们要最小化平方和,也即平均分,具体来说,有 \displaystyle x-S\bmod x 组有 \displaystyle\lfloor\frac{S}{x}\rfloor 人,\displaystyle S\bmod x 组有 \displaystyle\lfloor\frac{S}{x}\rfloor+1 人,贡献可以 O(1) 计算。
问题在于要枚举两维。我们发现 x 这一维省不掉,因为要减去 (x-1)X;但是注意到人数总和不超过 V=2\times 10^5,由根号结论可知本质不同的 S 只有 O(\sqrt V) 种,于是暴力枚举即可。