[蓝桥杯][2015年第六届真题]机器人塔(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
思路：我们从顶开始往下分析，对于这一行的字符串排列顺序，如果开头确定了，那么后面的随之也就确定了，因此分类讨论就可以了。
代码如下：

#include<bits/stdc++.h>
#define ll long long
using namespace std;

int n,m;

inline ll dfs(int a,int b,int len,string s)
{
	if(a==0&&b==0) return 1ll;
	if(len==0) return dfs(a-1,b,1,"A")+dfs(a,b-1,1,"B");
	else
	{
		string t="";
		int c=a,d=b;
		ll sum=0;
		if(s[0]=='A') 
		{
			if(c>=2)
			{
				t+="AA";c-=2;
				for(int i=1;i<s.length();i++) 
				{
					if(s[i]=='A'&&t[i-1]=='A'&&c) t+='A',c--;
					else if(s[i]=='A'&&t[i-1]=='B'&&d) t+='B',d--;
					else if(s[i]=='B'&&t[i-1]=='A'&&d) t+='B',d--;
					else if(s[i]=='B'&&t[i-1]=='B'&&c) t+='A',c--;
				}
				if(t.length()==len+1) sum+=dfs(c,d,len+1,t);
				c=a,d=b,t="";
			}
			if(d>=2)
			{
				t+="BB";d-=2;
				for(int i=1;i<s.length();i++) 
				{
					if(s[i]=='A'&&t[i-1]=='A'&&c) t+='A',c--;
					else if(s[i]=='A'&&t[i-1]=='B'&&d) t+='B',d--;
					else if(s[i]=='B'&&t[i-1]=='A'&&d) t+='B',d--;
					else if(s[i]=='B'&&t[i-1]=='B'&&c) t+='A',c--;
				}
				if(t.length()==len+1) sum+=dfs(c,d,len+1,t);
				c=a,d=b,t="";
			}
		}
		c=a,d=b,t="";
		if(s[0]=='B')
		{
			if(c&&d)
			{
				t+="AB";c--,d--;
				for(int i=1;i<s.length();i++) 
				{
					if(s[i]=='A'&&t[i-1]=='A'&&c) t+='A',c--;
					else if(s[i]=='A'&&t[i-1]=='B'&&d) t+='B',d--;
					else if(s[i]=='B'&&t[i-1]=='A'&&d) t+='B',d--;
					else if(s[i]=='B'&&t[i-1]=='B'&&c) t+='A',c--;
				}
				if(t.length()==len+1) sum+=dfs(c,d,len+1,t);
				c=a,d=b,t="";
				t+="BA";c--,d--;
				for(int i=1;i<s.length();i++) 
				{
					if(s[i]=='A'&&t[i-1]=='A'&&c) t+='A',c--;
					else if(s[i]=='A'&&t[i-1]=='B'&&d) t+='B',d--;
					else if(s[i]=='B'&&t[i-1]=='A'&&d) t+='B',d--;
					else if(s[i]=='B'&&t[i-1]=='B'&&c) t+='A',c--;
				}
				if(t.length()==len+1) sum+=dfs(c,d,len+1,t);
				c=a,d=b,t="";
			}
		}
		return sum;
	}
}
int main()
{
	while(~scanf("%d%d",&n,&m))
	{
		printf("%lld\n",dfs(n,m,0,""));
	}
	return 0;
}

努力加油a啊，(^o)/~