AcWing 1321. 取石子(博弈dp)

最新推荐文章于 2024-09-25 11:53:45 发布

做一只大熊猫

最新推荐文章于 2024-09-25 11:53:45 发布

阅读量451

点赞数 1

分类专栏：博弈文章标签：算法图论深度优先

本文链接：https://blog.csdn.net/qq_34261446/article/details/121531535

版权

博弈专栏收录该内容

6 篇文章 0 订阅

订阅专栏

在这里插入图片描述

代码

#include <bits/stdc++.h>
using namespace std;
#define db  double
#define ll  long long
#define Pir pair<int, int>
#define fi  first
#define se  second
#define pb  push_back
#define m_p make_pair
#define inf 0x3f3f3f3f
#define INF 0x3f3f3f3f3f3f3f3f
/*==========ACMer===========*/
const int N = 55, M = 50005;
int f[N][M];

int dp(int a, int b)
{
    int & v = f[a][b];
    if (~ v) return v;
    if (! a) return b % 2;
    if (b == 1) return dp(a + 1, 0);

    if (a && ! dp(a - 1, b)) return v = 1;
    if (b && ! dp(a, b - 1)) return v = 1;
    if (a >= 2 && ! dp(a - 2, b + (b ? 3 : 2))) return v = 1;
    if (a && b && ! dp(a - 1, b + 1)) return v = 1;

    return v = 0;
}

int main()
{
    memset(f, -1, sizeof f);
    int T; scanf("%d", &T);
    while (T --)
    {
        int n; scanf("%d", &n);
        int a = 0, b = 0;
        for (int i = 0; i < n; i ++) {
            int x; scanf("%d", &x);
            if (x == 1) a ++;
            else b += b ? x + 1 : x;
        }

        if (dp(a, b)) printf("YES\n");
        else printf("NO\n");
    }

    return 0;
}