1986E-Beautiful Array

题目链接：Beautiful Array

我们可以从两个数开始考虑，什么条件下符合呢？那也就是当两个数%k都是相同的时候成立，因此我们可以采用map数组存储a[i]%k的值，套一个vector数组实现，当n为奇数时，意味着其中一个数可以不用匹配，偶数时必须匹配，因此我们可以查找每个%k的数的奇数的个数来判断是否符合条件。

接下来就考虑实现过程，当x%k这个数组的个数有偶数个，那么不用思考，直接将相邻的两个差的绝对值相加就得到了，那么如果是奇数的呢？我们可以采用前缀和和后缀和的思想去解决，具体实现看代码：

#include <bits/stdc++.h>

using namespace std;
map<long long, vector<long long> > m;
long long n, k, ans, cnt;
void solve() {
    cin >> n >> k;
    for (int i=1; i<=n; i++) {
        long long x;
        cin >> x;
        m[x%k].push_back(x/k);
    }
    for (auto p:m) {
        int si=p.second.size();
        sort(p.second.begin(), p.second.end());
        if (si%2) {
            cnt++;
            if (si==1) continue;
            vector<long long> a, b;
            for (int i=1; i<si; i++) {
                long long val=p.second[i]-p.second[i-1];
                if (i%2) a.push_back(val);
                else b.push_back(val);
            }
            int sa=a.size();
            for (int i=1; i<sa; i++) {
                a[i]+=a[i-1];
            }
            for (int i=sa-2; i>=0; i--) {
                b[i]+=b[i+1];
            }
            long long mini=min(a[sa-1], b[0]);
            for (int i=0; i<sa-1; i++) {
                mini=min(mini, a[i]+b[i+1]);
            }
            ans+=mini;
        }
        else {
            for (int i=1; i<si; i+=2) {
                ans+=p.second[i]-p.second[i-1];
            }
        }
    }
    if (cnt>1) {
        cout << -1 << "\n";
    } else {
        cout << ans << "\n";
    }
    ans=0, cnt=0;
    m.clear();
}
int main()
{
    ios_base::sync_with_stdio(false);
    int w;
    cin >> w;
    while (w--) {
        solve();
    }
    return 0;
}