机器人塔
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
请严格按要求输出,不要画蛇添足地打印类似:“请您输入…” 的多余内容。
所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。
注意:不要使用package语句。不要使用jdk1.7及以上版本的特性。
注意:主类的名字必须是:Main,否则按无效代码处理。
import java.util.Scanner;
public class Main {
public static int m, n;
public static int[][] value; //此处用数字1代表A,数字2代表B
public static int len, count = 0;
public void check() {
for(int i = len - 2, t = 1;i >= 0;i--, t++)
for(int j = 0;j < len - t;j++) {
if(value[i + 1][j] == value[i + 1][j + 1])
value[i][j] = 1;
else
value[i][j] = 2;
}
int sumA = 0, sumB = 0;
for(int i = 0, t = len - 1;i < len;i++, t--)
for(int j = 0;j < len - t;j++) {
if(value[i][j] == 1)
sumA++;
else if(value[i][j] == 2)
sumB++;
}
if(sumA == m && sumB == n)
count++;
}
public void dfs(int step) {
if(step == len) {
check();
return;
} else {
for(int i = 1;i <= 2;i++) {
value[len - 1][step] = i; //确定三角形最底层一行,即可确定整个三角形
dfs(step + 1);
}
}
}
public static void main(String[] args) {
Main test = new Main();
Scanner in = new Scanner(System.in);
m = in.nextInt();
n = in.nextInt();
for(int i = 1;i < 100;i++) { //满足三角形规则,可知i层,则(i + 1) * i / 2 = m + n
int a = (i + 1) * i / 2;
if(a == m + n) {
len = i;
value = new int[a][a];
break;
}
}
test.dfs(0);
System.out.println(count);
}
}