蓝桥杯取球游戏_博弈入门(附加动态规划).java

今盒子里有n个小球，A、B两人轮流从盒中取球，每个人从盒子中取出的球的数目必须是：1，3，7或者8个。
    A先取球，然后双方交替取球，直到取完。
    被迫拿到最后一个球的一方为负方（输方）
    请编程确定出在双方都不判断失误的情况下，对于特定的初始球数，A是否能赢？
    程序运行时，从标准输入获得数据，其格式如下：
    先是一个整数n(n<100)，表示接下来有n个整数。然后是n个整数，每个占一行（整数<10000），表示初始球数。
    程序则输出n行，表示A的输赢情况（输为0，赢为1）。
    例如，用户输入：
４
１
２
10
18
    则程序应该输出：
0
1
1
0

博主只写了输入整数，判断这个球数是否可获胜，感兴趣按照题意进行补充。

import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;

public class TakeBall {
	//保存一定胜利的球数
	static Set<Integer> win = new HashSet<Integer>();
	//保存一定失败的球数
	static Set<Integer> lose = new HashSet<Integer>();
	public static void main(String[] args) {
		/*
		 * 测试后发现不能直接计算10000，会导致栈溢出
		 * 先缓存一次7000，再计算10000即可
		 * 如果需要计算更大数据 先缓存小数据即可
		 */
		f(7000);
		Scanner sc = new Scanner(System.in);
		while(true){
			if(f(sc.nextInt())){
				System.out.println("赢");
			}else{
				System.out.println("输");
			}
		}
	}
	static boolean f(int n){
		//如果动态数组中有这个n直接返回即可
		if(win.contains(n))	return true;
		if(lose.contains(n)) return false;
		//依次遍历，当我取1,3,7,8个球时，对方是否会获胜
		//对方失败，我胜利，并将胜利数据加入动态数组中，可直接调用
		if(n>1&&!f(n-1)){ win.add(n); return true;}
		if(n>3&&!f(n-3)){ win.add(n); return true;}
		if(n>7&&!f(n-7)){ win.add(n); return true;}
		if(n>8&&!f(n-8)){ win.add(n); return true;}
		//对方胜利，我失败，并将失败数据加入动态数组中，可直接调用
		lose.add(n);
		return false;
	}
}

有任何问题欢迎交流学习。