题目描述
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<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<vector>
#include<cmath>
const int maxn=1e5+5;
typedef long long ll;
using namespace std;
int a[50][50];
int k=1;
int vis[55];
int s=0;
int n,m;
void dfs(int x)
{
if(x==k)
{
for(int j=k-2;j>=0;j--)
{
for(int kk=0;kk<=j;kk++)
if((a[j+1][kk]==0&&a[j+1][kk+1]==1)||(a[j+1][kk]==1&&a[j+1][kk+1]==0))
{
a[j][kk]=1;
}
else
{
a[j][kk]=0;
}
}
int s1=0;
int s2=0;
for(int t=0;t<k;t++)
{
for(int j=0;j<=t;j++)
{
if(a[t][j]==0)
{
s1++;
}
else
{
s2++;
}
}
}
if(s1==n&&s2==m)
{
s++;
}
return ;
}
if(vis[x]==0)
{
vis[x]=1;
a[k-1][x]=1;
dfs(x+1);
a[k-1][x]=0;
dfs(x+1);
vis[x]=0;
}
}
int main()
{
cin>>n>>m;
int sum=0;
while(sum+k<n+m)
{
sum+=k;
k++;
}
dfs(0);
cout<<s<<endl;
return 0;
}