机器人塔蓝桥杯-java

机器人塔
X星球的机器人表演拉拉队有两种服装，A和B。
他们这次表演的是搭机器人塔。

类似：

 A
B B

A B A
A A B B
B B B A B
A B A B B A
队内的组塔规则是：

A 只能站在 AA 或 BB 的肩上。
B 只能站在 AB 或 BA 的肩上。

你的任务是帮助拉拉队计算一下，在给定A与B的人数时，可以组成多少种花样的塔。

输入一行两个整数 M 和 N，空格分开（0<M,N<500）,分别表示A、B的人数，保证人数合理性。

要求输出一个整数，表示可以产生的花样种数。

例如：
用户输入：
1 2

程序应该输出：
3

再例如：
用户输入：
3 3

程序应该输出：
4

资源约定：
峰值内存消耗 < 256M
CPU消耗 < 1000ms

分析：
分两种情况，如果我们从上面开始往下面走，这时候，每走一步需要判断下一步是需要放A还是放B，在每一步的时候都有两种选择，第二种情况，我们从最底下往上面分析，我们只需要将最后一层安排好之后，上面的层数一定是固定的，因为A和A，B和B的上面只能放A，A和B的上面只能放B，这样的话，每一步我们只需要判断当前如果这一步放A或者放B的时候会不会发生A或者B不够的问题。因此选择第二种走法。
我们将A当做0，B当做1，这时候我们只需要将最下面一层的情况尽数遍历一下就行，也就是从0000000…到1111111111，即全零到全一的所有情况。
重点：在自下而上的时候，我们判断每一点的时候，如果使用if else的话会看着比较繁琐，这时候一个巧妙的点就是使用异或运算，如果 0 和 1 异或是1 ，1和1， 0 和 0 异或的结果都是零，这样的话效率会高很多。

源码：

package 第七届C语言B组2;

import java.util.Scanner;

public class 机器人塔 {
	public static int shu[][];
	public static int a,b,num = 0;
	public static void main(String[] args) {
		Scanner input= new Scanner(System.in);
		a = input.nextInt();
		b = input.nextInt();
		int all = (a+b)*2;
		int t = 1;
		for (int i = 1; i < 100000; i++) {
			if (i*i+i == all) {
				t =i;
				break;
			}
		}
		shu = new int[t][t];
		dfs(0);
		System.out.println(num);
	}
	public static void dfs(int index) {
		if (index == shu.length) {//现在已经完成了一种状态
			if (check()) {
				num++;
			}
			return;
		}
		int next[]= {0,1};
		int tx;
		for (int i = 0; i < next.length; i++) {
			tx = next[i];
			shu[shu.length-1][index] = tx;
			dfs(index+1);
			shu[shu.length-1][index] = 0;
		}
	}
	public static boolean check() {
		int count[] =new int[2];//0代表a，1代表b
		for (int i = shu.length-1; i < shu.length; i++) {
			for (int j = 0; j < shu.length; j++) {
				count[shu[i][j]]++;
				if (count[0] > a || count[1] > b) {
					return false;//说明此时已经不够了
				}
			}
		}
		for(int i = shu.length-2;i >= 0;i--) {
			for (int j = 0; j <= i; j++) {
				shu[i][j] = shu[i+1][j]^shu[i+1][j+1];
				count[shu[i][j]]++;
				if (count[0] > a || count[1] > b) {
					return false;//说明此时已经不够了
				}
			}
		}
		if (count[0] == a && count[1] == b)  {
			return true;
		}else {
			return false;
		}
	}
}