素敵な宝箱 - yukicoder No. 1060

立式する.


私が i 番目の宝石を合計 x 個とるとする.
i 番目の宝石は計 Si 個あるとする.
X - Y は,  x^2 - (S_{i} - x)^2 = 2 S_{i} x - S_{i}^2
(x =  A_{1'i} + ... + A_{k'i})


宝箱ごとに, すべての宝石 i に関して 2 Si x の合計を求める.
私は, 大きい順に 1 つとばしで得る.


最後に, すべての宝石 i に関して Si^2 を引く.


ソースコード
https://yukicoder.me/submissions/486237