Submission #1232463

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const double pi = acos(-1);
#define EACH(i,a) for (auto& i : a)
#define FOR(i,a,b) for (ll i=(a);i<(b);i++)
#define RFOR(i,a,b) for (ll i=(b)-1;i>=(a);i--)
#define REP(i,n) FOR(i,0,n)
#define RREP(i,n) RFOR(i,0,n)
template<typename T> ostream& operator<<(ostream& os, const vector<T>& v) {
REP(i,v.size()){if(i)os<<" ";os<<v[i];}return os;}
template<typename T> ostream& operator<<(ostream& os, const vector<vector<T>>& v) {
REP(i,v.size()){if(i)os<<endl;os<<v[i];}return os;}

bool solve(vector<ll> A) {
    ll n = A.size();
    // ΣA[i] を求める
    ll sum = 0;
    REP(i, n) {
        sum += A[i];
    }
    // 1回の操作で取り除く個数
    ll K = n * (n+1) / 2;
    // 割り切れないとどう頑張ってもちょうど全部取り除けない
    if (sum % K != 0) {
        return false;
    }
    // 合計操作回数
    ll T = sum / K;
    REP(i, n) {
        // 隣との差を埋められるか
        ll d = A[(i+1)%n] - A[i] - T;
        if (d > 0 || -d % n != 0) {
            return false;
        }
    }
    return true;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    // メインプログラムここから
    ll n; cin >> n;
    vector<ll> A(n);
    REP(i, n) {
        cin >> A[i];
    }
    if (solve(A)) {
        cout << "YES" << endl;
    }
    else {
        cout << "NO" << endl;
    }
}

Submission Info

Judge Result