题解 | 元素方碑

#include <iostream>
#include<vector>
using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    int t, n;
    cin >> t;
    while (t--) {
        cin >> n;
        long long sum = 0;
        vector<int> vec(n);
        for (int i = 0; i < n; i++) {
            cin >> vec[i];
            sum += vec[i];
        }
        if (sum % n != 0) {
            cout << "NO\n";
            continue;
        }
        long long av = sum / n;
        for (int i = 1; i < n - 1; i++) {
            vec[i + 1] += vec[i - 1] - av;
            vec[i - 1] = av;
        }
        if (vec[n - 1] != av)
            cout << "NO\n";
        else
            cout << "YES\n";
    }
}
// 64 位输出请用 printf("%lld")