【DFS+记忆化搜索】 蓝桥杯真题 2016初赛 取球博弈

题目描述

两个人玩取球的游戏。一共有N个球，每人轮流取球，每次可取集合{n1,n2,n3}中的任何一个数目。
如果无法继续取球，则游戏结束。此时，持有奇数个球的一方获胜。如果两人都是奇数，则为平局。
假设双方都采用最聪明的取法，第一个取球的人一定能赢吗？试编程解决这个问题。

输入
输入存在多组测试样例，对于每一组测试数据：
第一行3个正整数n1 n2 n3，空格分开，表示每次可取的数目 (0<n1,n2,n3<100)
第二行5个正整数x1 x2 … x5，空格分开，表示5局的初始球数(0<xi<1000)

输出
一行5个字符，空格分开。分别表示每局先取球的人能否获胜。
能获胜则输出+，次之，如有办法逼平对手，输出0，无论如何都会输，则输出-

样例输入

1 2 3
1 2 3 4 5
1 4 5
10 11 12 13 15
2 3 5
7 8 9 10 11

样例输出

+ 0 + 0 -
0 - 0 + +
+ 0 0 0 0

思路

说实话，一开始做这道题，一直wa没做出来，后面看了题解才过了=_=
ac思路：

1. 无须关心实际取了多少个球，只需要记录先手和后手最后拿球的奇偶数接可以了
2. 先计算先手是否能赢，若不能则考虑是否能平手，否则最后只能输
3. 由于计算过程中可能有很多情况已经计算过，可以使用记忆化数组将结果保存起来，若该种情况已经计算过，则直接返回结果
4. 详细思路见下面代码注释

代码

#include<bits/stdc++.h>
using namespace std;
int a[3];	//记录可取数目
int dp[1005][2][2] = {0};    //记忆化数组

// x为本次拿球方
int dfs(int num, int x, int y){
	//若无法继续拿球
    if(num<a[0]){
        if(x==1 && y==0)
            return dp[num][x][y] = 1;
        else if(x==0 && y==1)
            return dp[num][x][y] = 3;
        else
            return dp[num][x][y] = 2;
    }

	//若该种情况已计算过
    if(dp[num][x][y]!=0)
        return dp[num][x][y];

	//记录是否存在平局现象
    int flag = 0;

    for(int i=0; i<3; i++){
        if(num>=a[i]){
        	//交换拿球方，由y拿球，并计算x已拿球的奇偶数，返回对方那球结果
            int tmp = dfs(num-a[i], y, a[i]%2==0?x:(1-x));
			
			//若对方拿球后输了，本方就赢了
            if(tmp==3){
                return dp[num][x][y] = 1;
            }
		
			//若平局，则记录下来
            if(tmp==2)
                flag = 1;
        }
    }
	//若上面循环没有return结束，则证明本轮x拿球要么平局，那么输掉
    if(flag)
        return dp[num][x][y] = 2;
    else
        return dp[num][x][y] = 3;
}


int main(){


    while(cin>>a[0]>>a[1]>>a[2]){
        int x;
		
        memset(dp, 0, sizeof(dp));		//初始化记忆数组

        sort(a, a+3);	//对可拿球数排序，方便计算

        for(int i=0; i<5; i++){
            cin>>x;
            int ans = dfs(x, 0, 0);

            if(ans==1){
                cout<<"+ ";
            }else if(ans==3){
                cout<<"- ";
            }else{
                cout<<"0 ";
            }
        }
        cout<<endl;
    }

    return 0;
}