题解 | 元素方碑

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

int main() {
   int t;
   cin>>t;
   while(t--){
    int n;
    cin>>n;
    long long  a[1000000];
    for(int i=0;i<n;i++){
        cin>>a[i];
    }

    long long sum=0;
    for(int i=0;i<n;i++){
        sum=sum+a[i];
    }
	 //条件1：整体的平均值为整数
    if(sum % n!=0){
        cout<<"NO"<<endl;
        continue;
    }
    long long sum1=0;
    long long sum2=0;
    long long count=0;
	 //边界单独讨论，防止后面有除以0情况
    if(n==1){
        cout<<"YES"<<endl;
        continue;
    }
	 //题目中的条件表明：奇数位和偶数位的数字分别单独改变，直到相等，也就是说，奇数位和偶数位上的平均值必须都要和总平均值相等才可以
    for(int i=0;i<n;i++){
        if(i%2==0){
            sum1+=a[i];
            count++;
        }
        else{
            sum2+=a[i];
        }
    }
    if((sum1/count)==(sum2/(n-count))){
        cout<<"YES"<<endl;
    }
    else{
        cout<<"NO"<<endl;
    }
   }
return 0;
}
// 64 位输出请用 printf("%lld")