HDOJ 4155 The Game of 31 博弈搜索

//HDOJ 4155 The Game of 31 博弈搜索
/*
题意：有编号为1、2、3、4、5、6的牌各4张，共24张，两个人轮流取牌，
取牌后要使所有取出的牌的总不超过31，不能取的输。
游戏从给定的局面开始。

思路：从给定的局面开始dfs，当此状态能够到达p状态，则此状态有n状态
否则为p状态，然后判断此时是先手还是后手
*/

#include<stdio.h>
#include<string.h>
#include<stdlib.h>

#define N 100
#define M 10

char str
;
int num[M];
int len,sum;

void init(){
int i;
for(i = 1; i <= 6; ++i)
num[i] = 4;
len = strlen(str);
sum = 0;
for(i = 0; i < len; ++i){
--num[str[i]-'0'];
sum += str[i]-'0';
}
}

bool dfs(int sum){
int i;
if(sum > 31)
return 0;
for(i = 1; i <= 6; ++i){
if(num[i] && sum+i<=31){
--num[i];
if(dfs(sum+i) == 0){
++num[i];
return 1;
}
++num[i];
}
}
return 0;
}

int main(){
while(scanf("%s",&str)!=EOF){
init();
printf("%s ",str);
if(dfs(sum))
puts(len%2 ? "B" : "A");
else
puts(len%2 ? "A" : "B");
}
return 0;
}