第七届蓝桥杯（国赛）——机器人塔

问题描述
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，空格分开，分别表示 A、B 的人数，保证人数合理性。

输出格式
要求输出一个整数，表示可以产生的花样种数。

样例输入1
1 2

样例输出1
3

样例输入2
3 3

样例输出2
4

数据范围
0 < M, N < 500

---

题解
DFS：

解题思路：每放置完一层机器人，就判断站法是否正确，当所有机器人都放置完毕后，再检测数量是否正确。

#include <iostream>
using namespace std;

int n, m, k = 1, ans;
char s[100][100];

bool cnt()												// 判断数量是否正确 
{
	int a = 0, b = 0;
	for (int i = 1; i <= k; i ++)
		for (int j = 1; j <= i; j ++)
			if(s[i][j] == 'A') a ++;
			else b ++;
	
	if(a == n && b == m) return true;
	return false;		
}

bool judge(int x)										// 判断站法是否正确 
{
	for (int i = 1; i < x; i ++)
		for (int j = 1; j <= i; j ++)
		{
			if(s[i][j] == 'A')
			{
				if(s[i + 1][j] == 'A' && s[i + 1][j + 1] == 'B') return false;
				if(s[i + 1][j] == 'B' && s[i + 1][j + 1] == 'A') return false; 
			}
			
			if(s[i][j] == 'B')
			{
				if(s[i + 1][j] == 'A' && s[i + 1][j + 1] == 'A') return false;
				if(s[i + 1][j] == 'B' && s[i + 1][j + 1] == 'B') return false; 
			}	
		}
		
	return true;	
}

void dfs(int x, int y)
{			
	if(y == x + 1)
	{
		if(judge(x)) x ++, y = 1;					// 换行 
		else return;
	}
	
	if(x == k + 1)
	{ 
		if(cnt()) ans ++;							// 方案数 +1 
		return;
	}
	
	for (char i = 'A'; i <= 'B'; i ++)
	{
		s[x][y] = i;                                // 放置机器人 
		dfs(x, y + 1);
	}
}

int main()
{
	cin >> n >> m;
	
	while((1 + k) * k / 2 != n + m) k ++;			// 获取层数 

	dfs(1, 1);
			
	cout << ans << endl;
	return 0;	
}

ps：感觉我这种方法比较简单，但是调试了好久，还是太弱🤢