codeforces 768E. Game of Stones

题面

题意

在Nim取石子的游戏上作了升级,对于同一堆石子,如果曾经取了k个,则不能再取k个,不能再取的人输,取完的人赢,判断后手是否能赢.

做法

如果去掉限制条件,那么这只要Xor所有石子的个数即可,现在其实可以将题目转化为原来的题目,对于一堆石子可以求出它最多能被取几次,比如11个石子最多能被取4次,那么它就可以被看成4个石子,因为可以将11个变成最多还能取0,1,2,3次的石子,将转化后的石子数Xor一下即可.

代码

#include<iostream>
#include<cstdio>
using namespace std;

int n,ans;

int main()
{
    int i,j,p;
    cin>>n;
    for(i=1;i<=n;i++)
    {
        scanf("%d",&p);
        for(j=1;j*(j+1)/2<=p;j++);
        ans^=j-1;
    }
    ans?puts("NO"):puts("YES");
}