AGC010 - B: Boxes

时间:2023-03-09 15:36:10
AGC010 - B: Boxes

原题链接

题意简述

给出一个由n(n≤105)个数构成的环,每次可以选择一个位置并从这个数起顺时针依次对每个数-1,-2,-3,…,-n。问能否将所有数全变为0。

分析

考虑一次操作对环带来了什么影响。

(在an后加一个a1来表示数环)

6,3,5,7,9(,6) -> 5,1,2,3,4(,5)

5,1,2,3,4(,5) -> 0,0,0,0,0(,0)

差分后:

−3,2,2,2,−3 -> −4,1,1,1,1

−4,1,1,1,1 -> 0,0,0,0,0

可以看到,一次操作相当于对差分数列(或者说是差分环)的一个位置加上n-1,剩下的位置减去1。那么只要检查原环的差分数列能否全变为0,并且此时和也为0就行了。

对每一个位置的计算复杂度为O(1),总时间复杂度为O(n)。

实现

每次操作会使和sum减少 s0=n(n+1)/2,那么总共进行了 k=sum/s0 次操作。如果k不为整数那么不可行。

差分数列的每个位置要能在数个 +(n−1) 和 −1 后变为0,否则不可行。

列式表示为 (ai+1−ai)+xi(n−1)−(k−xi)=0,如果任何一个xi不为整数那么不可行。

最后,如果∑xi≠k说明此时sum≠0,不可行。

代码

#include <cstdio>
typedef long long lint;
int const N=1e5+10;
int n,a[N];
int dif[N];
int main()
{
freopen("b.in","r",stdin);
scanf("%d",&n);
if(n==1) {printf("YES"); return 0;}
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
a[n+1]=a[1];
lint s=(lint)n*(n+1)>>1,sum=0;
for(int i=1;i<=n;i++) sum+=a[i],dif[i]=a[i+1]-a[i];
if(sum%s!=0) {printf("NO"); return 0;}
lint k=sum/s,sumX=0;
for(int i=1;i<=n;i++)
{
lint x=(k-dif[i])/n;
if(x<0 || x*n!=k-dif[i]) {printf("NO"); return 0;}
sumX+=x;
}
if(sumX==k) printf("YES");
else printf("NO");
return 0;
}

注意

开longlong!int*int也有可能爆int,要先转成longlong再乘!

连WA六发…