机器人塔

机器人塔

时间限制: 1Sec 内存限制: 128MB 提交: 26 解决: 18
题目描述
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的人数，保证人数合理性。

要求输出一个整数，表示可以产生的花样种数。
输入
输入一行两个整数 M 和 N，空格分开（0<M,N<500）,分别表示A、B的人数，保证人数合理性。
输出
要求输出一个整数，表示可以产生的花样种数。
样例输入

1 2

样例输出

3

思路：确定最下面的一行 向上递推

#include<iostream>
#include<algorithm>
using namespace std;
char b[50];
char a[100];
int M, N;
int num;
int ans = 0;
void judge()
{
 int x = num;
 int m = 0, n = 0;
 for(int i = 0; i < x; i++){
     b[i] = a[i];
  if(a[i] = 'A'){
       n++;
  } 
  else  {
       m++;
  }         
 }
 while(x--){
  for (int i = 0;i < x; i++){
   if (b[i] == b[i+1]){
    n++;
    b[i] = 'A';
   }
   else{
    m++;
    b[i] = 'B';
   }
  }
 }
 if(m == M && n == N){
  ans++;
 }
}
void dfs(int x)
{
 if(x == num)
 {
    judge();
    return ;
 }
 for (char i='A';i<='B';i++)
    {
        a[x]=i;
        dfs(x+1);
    }
}
int main()
{
 cin >> M >> N;
 for(int i = 1; i < 500; i++)
    {
        if ((i + 1) * (i)/2 == N + M)
        {
            num = i;
            break;
        }
    }
    dfs(0);
    cout << ans;
    return 0;
}