Problem Statement
There are N boxes arranged in a circle. The i-th box contains Ai stones.
Determine whether it is possible to remove all the stones from the boxes by repeatedly performing the following operation:
Select one box. Let the box be the i-th box. Then, for each j from 1 through N, remove exactly j stones from the (i+j)-th box. Here, the (N+k)-th box is identified with the k-th box.
Note that the operation cannot be performed if there is a box that does not contain enough number of stones to be removed.
Constraints
1≦N≦105
1≦Ai≦109
Input
The input is given from Standard Input in the following format:
N
A1 A2 … AN
Output
If it is possible to remove all the stones from the boxes, print YES. Otherwise, print NO.
Sample Input 1
5
4 5 1 2 3
Sample Output 1
YES
All the stones can be removed in one operation by selecting the second box.
Sample Input 2
5
6 9 12 10 8
Sample Output 2
YES
Sample Input 3
4
1 2 3 1
Sample Output 3
NO
这题当时看没有看懂,就一直撂在一遍,现在重新拾起来,至今大部分的题解我还没有看懂。
这是我第一次见到用线性代数的思维来做的题目。还是太年轻,做题太少。
题目大意:就是说有n个盒子,每个盒子里面有ai块石头。可以选择第i个盒子,然后往后依次在第(i+j)%n个盒子中取走j块石头,1<=j<=n,如果盒子里的石头不够则不能进行操作。问能否把石头全部取走。
思路:
取走石头的方式无非就n种,也就是说选择第i个盒子的方式有n种,假设每种有xi次操作。
那么就有方程组可得。写成矩阵形式
在这里面规定好了只能拿走石头,而不能放石头在里面,所以对于每一个xi,都不可能是小于0或者是小数的。
看起来这个方程很大,但是并不难解。
首先将每一个方程项加起来得:
(1+2+3+…+n)*(x1+x2+…+xn)=a1+a2+a3+…+an.
设x1+x2+x3+…+xn=s, a1+a2+a3+…+an=sum;
那么就得到s=sum/(n*(n+1)/2);(这里sum必须是整除的,否则没有必要继续计算下去,直接输出no)
然后相邻的方程相减,比如第一个减去第二个得到
(n-1)*x1-x2-x3-…-xn=a1-a2;
-s+n*x1=a1-a2;
x1=(s+a1-a2)/n;
这样就能解出x1了,这里要判断条件,如果x1是负数或者是小数,那么当然就不能达成目的,自然输出no,否则输出yes。
下面是代码:要用long long
#include<iostream>
#include<cstdio>
#include<cmath>
typedef long long ll;
using namespace std;
ll n;
ll a[100010];
ll dis[100010];
int main()
{
ll sum=0;
scanf("%lld",&n);
for(int i=0;i<n;i++)
{
scanf("%lld",&a[i]);
sum+=a[i];
}
ll k;
if((2*sum%(n*(n+1)))!=0)
{
printf("NO");
return 0;
}
k=2*sum/n/(n+1);
for(int i=0;i<n;i++)
{
if(i==n-1)
dis[i]=a[i]-a[0];
else
dis[i]=a[i]-a[i+1];
}
for(int i=0;i<n;i++)
{
if(dis[i]+k<0||(dis[i]+k)%n!=0)
{
printf("NO");
return 0;
}
}
printf("YES");
return 0;
}