蓝桥杯取球博弈

题目描述

两个人玩取球的游戏。
一共有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

程序应该输出：
+ 0 + 0 -

再例如，输入：
1 4 5
10 11 12 13 15

程序应该输出：
0 - 0 + +

再例如，输入：
2 3 5
7 8 9 10 11

程序应该输出：
+ 0 0 0 0

资源约定：
峰值内存消耗（含虚拟机） < 256M
CPU消耗 < 3000ms

暴力法

暴力法，遍历所有可能情况，效率较低

import java.util.*;

public class Main {

	static int[] n = new int[3];

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		for (int i = 0; i < n.length; i++) {
			n[i] = sc.nextInt();
		}
		for (int i = 0; i < 5; i++) {
			int num = sc.nextInt();
			char res = f(num, 0, 0);
			System.out.print(res + " ");
		}
		System.out.println();
		sc.close();
	}

	/**
	 * 遍历所有可能的情况，从中找出最好的情况输出
	 * 
	 * @param num
	 *            当前未取剩余球数
	 * @param myNum
	 *            我方手中球数
	 * @param yourNum
	 *            对方手中球数
	 * @return
	 */
	private static char f(int num, int myNum, int yourNum) {
		// 剩下的num不够取
		if (num < n[0] && num < n[1] && num < n[2]) {
			if (myNum % 2 == 1 && yourNum % 2 == 0)
				return '+';
			else if (yourNum % 2 == 1 && myNum % 2 == 0)
				return '-';
			else
				return '0';
		}
		// 递归处理
		char[] res = new char[3];
		for (int i = 0; i < n.length; i++) {
			if (num >= n[i]) {
				res[i] = f(num - n[i], yourNum, myNum + n[i]);
				// res[i]是敌我互换身份以后的值
				// 敌方输了说明我方赢了，地方赢了说明我方输了，要将res[i]对调
				if (res[i] == '+') {
					res[i] = '-';
				} else if (res[i] == '-') {
					res[i] = '+';
				} else {
					res[i] = '0';
				}
			}
		}
		// return结果
		for (int i = 0; i < res.length; i++) {
			if (res[i] == '+') {
				return res[i];
			}
		}
		for (int i = 0; i < res.length; i++) {
			if (res[i] == '0') {
				return res[i];
			}
		}
		return '-';
	}
}

记忆型递归

相比上一次的代码，如果我们第一次求得剩余球数、敌我双方奇偶数情况下的结果，我们就将该结果存起来，下次要用时直接调用即可，而不用再去递归，来提升效率。

import java.util.*;

public class Main {

	static int[] n = new int[3];
	static char[][][] cache = new char[1000][2][2]; // 缓存剩余i张牌时的结果

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		for (int i = 0; i < n.length; i++) {
			n[i] = sc.nextInt();
		}
		for (int i = 0; i < 5; i++) {
			int num = sc.nextInt();
			char res = f(num, 0, 0);
			System.out.print(res + " ");
		}
		System.out.println();
		sc.close();
	}

	/**
	 * 遍历所有可能的情况，从中找出最好的情况输出
	 * 
	 * @param num
	 *            当前未取剩余球数
	 * @param myNum
	 *            我方手中球的奇偶性，偶为0，奇为1
	 * @param yourNum
	 *            对方手中球的奇偶性，偶为0，奇为1
	 * @return
	 */
	private static char f(int num, int myNum, int yourNum) {
		// 剩下的num不够取，将其缓存起来并返回相应的值
		if (num < n[0] && num < n[1] && num < n[2]) {
			if (myNum % 2 == 1 && yourNum % 2 == 0) {
				return cache[num][myNum][yourNum] = '+';
			} else if (yourNum % 2 == 1 && myNum % 2 == 0) {
				return cache[num][myNum][yourNum] = '-';
			} else {
				return cache[num][myNum][yourNum] = '0';
			}
		}
		// 如果当前的情况已经缓存，则直接返回缓存值
		if (cache[num][myNum][yourNum] != '\0') {
			return cache[num][myNum][yourNum];
		}
		// 递归处理
		char[] res = new char[3];
		for (int i = 0; i < n.length; i++) {
			if (num >= n[i]) {
				// 相对比上个版本的代码，这里将形参改变，因为我们其实只关心双方球数的奇偶性，而不管到底有多少个球
				// 当n[i]为偶数时，myNum的奇偶性不变，否则myNum改变奇偶性
				res[i] = f(num - n[i], yourNum, (n[i] % 2) == 0 ? myNum : 1 - myNum);
				// res[i]是敌我互换身份以后的值
				// 敌方输了说明我方赢了，敌方赢了说明我方输了，要将res[i]对调
				if (res[i] == '+') {
					res[i] = '-';
				} else if (res[i] == '-') {
					res[i] = '+';
				}
				// 如果res[i]为'0'，则不变
			}
		}
		// 将当前情况缓存到cache中，并return结果
		for (int i = 0; i < res.length; i++) {
			if (res[i] == '+') {
				// 将当前情况的结果缓存到cache中
				cache[num][myNum][yourNum] = res[i];
				return res[i];
			}
		}
		for (int i = 0; i < res.length; i++) {
			if (res[i] == '0') {
				// 将当前情况的结果缓存到cache中
				cache[num][myNum][yourNum] = res[i];
				return res[i];
			}
		}
		// 将当前情况的结果缓存到cache中
		cache[num][myNum][yourNum] = '-';
		return '-';
	}
}