Burger optimization

#include <bits/stdc++.h>
using namespace std;

void fast_io() { ios_base::sync_with_stdio(false); cin.tie(0); }

int main () {
    fast_io();

    int t;
    cin >> t;

    for(int i = 0; i < t; i++) {
        int k;
        cin >> k;
        vector<int> d(k);
        for(int j = 0; j < k; j++) {
            cin >> d[j];
        }

        sort(d.begin(), d.end());

        int s = 0;
        int sum = 0;
        for(int j = 0; j < k; j++) {
            sum += (d[j] - s) * (d[j] - s);
            if(j % 2 != 0)
                s++;
        }
        cout << sum << endl;
    }

    return 0;
}