蓝桥杯——机器人塔（dfs）

题目描述

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

先dfs最底下一层，然后再挨着判断上边每一层 

#include<cstdio>
#include<iostream>
#include<cstring>
#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,b,m,ans=0;
int pre[505];

bool check(){
	int num1=a;
	int num2=b;
	for(int i=1;i<=m;i++){
		if(pre[i]==1)
		    num1--;
		else
		    num2--;
	}
	for(int i=m-1;i>=1;i--){
		for(int j=1;j<=i;j++){
			if(pre[j]==pre[j+1])
			    pre[j]=1;
			else
			    pre[j]=0;
		}
		for(int j=1;j<=i;j++){
			if(pre[i]==1)
		        num1--;
		    else
		        num2--;
		    if(num1<0||num2<0)
		        return false;
		}
	}
	if(num1==0&&num2==0)
	return true;
}

void dfs(int tmp){
	if(tmp==m+1){
		if(check())
		    ans++;
		return ;
	}
	pre[tmp]=1;
	dfs(tmp+1);
	pre[tmp]=0;
	dfs(tmp+1);
}

int main(){
	scanf("%d%d",&a,&b);
	int n=a+b;
	int count=0;
	for(int i=1;i<50;i++){
		count+=i;
		if(count==n){
			m=i;//最底层机器人的数量 
			break;
		}
	} 
	dfs(1);
	printf("%d\n",ans);
    return 0;
}