【AtCoder】AGC010 Boxes

题目

传送门

题目大意

给出N(N105)N(N\leq 10^5)个数围成一圈,每次你可以选择一个i(1iN)i(1\leq i\leq N),然后对于每个j(1jN)j(1\leq j\leq N),从第i+ji+j个数(循环计数)中减去jj(若其中的某个数会减成负数,你就不能选择这个ii),问最后是否可能使每个数都变成00

思路

原题的题意就是从一个数开始,顺时针方向第ii个数减去ii(相当于一个等差数列)。

首先,所有数之和sumsum必须是1+2+...+N1+2+...+NN(N+1)2\dfrac{N(N+1)}{2}的倍数。
然后,进行的操作数k=sum1+2+...+Nk=\dfrac{sum}{1+2+...+N}

假设选了数aia_i,那么除了aia_iai+1a_{i+1}(都是循环计数),其他的每两个相邻的数之差都减小了11
例如:6,9,12,10,86,9,12,10,8,从66开始完成依一次这样的操作:
【AtCoder】AGC010 Boxes
于是,除了6699,其他的两个数之差都减小11,由于减去的是一个等差数列,所以不难理解这一点。
再看ai=6a_i=6ai+1=9a_{i+1}=9之差,增加了44,于是直接猜测:aia_iai+1a_{i+1}的差会增加N1N-1
事实上就是这样的,可以很容易地推出来。

发现差值dd最多减少kk,所以先判断dk>0d-k>0,若满足,任务不可能完成。
设进行了xx次“增加”的操作,则进行了(kx)(k-x)次“减小”的操作,一定有:d+x(N1)(kx)=0d+x(N-1)-(k-x)=0化简得:kd=xNk-d=xNkdk-d必须是NN的倍数。

做完了,脑袋痛。

代码

#include<set>
#include<cstdio>
#include<algorithm>
using namespace std;

#define MAXN 100000
int N;
int A[MAXN+5];

int main(){
    scanf("%d",&N);
    long long Sum=0;//注意long long
    for(int i=1;i<=N;i++){
        scanf("%d",&A[i]);
        Sum+=A[i];
    }
    long long All=1ll*N*(N+1)/2;
    if(Sum%All)
        return !puts("NO");
    int k=Sum/All;
    for(int i=1;i<=N;i++){
        int d=A[i]-A[i-1==0?N:i-1];
        if(d-k>0||(k-d)%N!=0)
            return !puts("NO");
    }
    puts("YES");
}