Educational Codeforces Round 77 (Rated for Div. 2) A. Heating

题目：https://codeforces.com/contest/1260/problem/A
思路：\(sum=\sum_{i=1}^{n}{a_i}{(n<=c)}\)
使 res=\(\sum_{i=1}^{n}{a_i^2}\) 最大 易得 \(max_a-min_a\leq1\)
设 \(val=sum/c\),\(cnt_{max}=sum%c\) 使\(cnt_{max}\)个a 值为val+1 其余为val 则最终值最大
\(res=cnt_{max} \times (val+1)^2+(c-cnt_{max}) \times val^2\)

#include

using namespace std;

int n;
int a[100];

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    //freopen("in.txt","r",stdin);
    int T;
    cin>>T;
    int c,sum;
    while(T--)
    {
        cin>>c>>sum;
        int val=sum/c;
        int cnt_max=sum%c;
        cout<