BZOj3895.取石子

Description

Alice和Bob两个好朋含友又开始玩取石子了。游戏开始时，有N堆石子
排成一排，然后他们轮流操作（Alice先手），每次操作时从下面的规则中任选一个：
·从某堆石子中取走一个
·合并任意两堆石子
不能操作的人输。Alice想知道，她是否能有必胜策略。

Input

第一行输入T，表示数据组数。
对于每组测试数据，第一行读入N。
接下来N个正整数a1，a2…an，表示每堆石子的数量。

Output

对于每组测试数据，输出一行。
输出YES表示Alice有必胜策略，输出NO表示Alice没有必胜策略。

Sample Input

2
3
1 1 2
2
3 4
3
2 3 5

Sample Output

YES
NO
NO

HINT

100%的数据满足T<=100, N<=50. ai<=1000

题解

考虑总的操作数，看这个数的奇偶性就可以了。
将石子堆分成两类：
1、石子数为1的。
2、石子数>1的。
对于第一类石子如果拿走了，就不能提供合并的操作。
是一个状态$f_{i,j}$表示第一类石子堆的数量为i，其余石子总和为j的胜败状态。
状态数很少，直接dfs就可以了。

code

#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#include <time.h>
#define ll long long
#define N 100003
#define M 103
#define db double
#define P putchar
#define G getchar
#define mo 998244353
#define inf 2551425  
using namespace std;
char ch;
void read(int &n)
{
    n=0;
    ch=G();
    while((ch<'0' || ch>'9') && ch!='-')ch=G();
    ll w=1;
    if(ch=='-')w=-1,ch=G();
    while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();
    n*=w;
}

int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
ll abs(ll x){return x<0?-x:x;}
ll sqr(ll x){return x*x;}
void write(ll x){if(x>9) write(x/10);P(x%10+'0');}

int T,n,m,s,a[N];
int f[53][50003];

int dfs(int x,int y)
{
    if(f[x][y]!=-1)return f[x][y];
    int p=214748367;
    if(x==0) return f[x][y]=y%2;
    if(y==1) return f[x][y]=dfs(x+1,0);

    if(x>1) p=min(p,dfs(x-2,y+2+(y?1:0)));
    if(x>=1) p=min(p,dfs(x-1,y));
    if(y>1) p=min(p,dfs(x,y-1));
    if(x && y) p=min(p,dfs(x-1,y+1));

    f[x][y]=p?0:1;

    return f[x][y];
}

int main()
{
    memset(f,-1,sizeof(f));
    for(read(T);T;T--)
    {
        read(n);
        m=s=0;
        for(int i=1;i<=n;i++)
        {
            read(a[i]);
            if(a[i]==1)m++;else s+=a[i]+1;
        }
        s=s?s-1:0;

        if(dfs(m,s))P('Y'),P('E'),P('S');
            else P('N'),P('O');
        P('\n');
    }

    return 0;
}