【思特奇杯·云上蓝桥-算法集训营】第1周 12-机器人塔

12-机器人塔

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

import java.util.Scanner;

public class code12_RobotTower {
    public static int arr[][];
    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;
            }
        }
        arr = new int[t][t];
        dfs(0);
        System.out.println(num);
    }
    public static void dfs(int index) {
        if (index == arr.length) {
            if (check()) {
                num++;
            }
            return;
        }
        int next[]= {0,1};
        int tx;
        for (int i = 0; i < next.length; i++) {
            tx = next[i];
            arr[arr.length-1][index] = tx;
            dfs(index+1);
            arr[arr.length-1][index] = 0;
        }
    }
    public static boolean check() {
        int count[] =new int[2];//0代表a，1代表b
        for (int i = arr.length-1; i < arr.length; i++) {
            for (int j = 0; j < arr.length; j++) {
                count[arr[i][j]]++;
                if (count[0] > a || count[1] > b) {
                    return false;//此时已经不够
                }
            }
        }
        for(int i = arr.length-2;i >= 0;i--) {
            for (int j = 0; j <= i; j++) {
                arr[i][j] = arr[i+1][j]^arr[i+1][j+1];
                count[arr[i][j]]++;
                if (count[0] > a || count[1] > b) {
                    return false;
                }
            }
        }
        if (count[0] == a && count[1] == b)  {
            return true;
        }else {
            return false;
        }
    }
}