【思特奇杯·云上蓝桥-算法训练营】第1周：机器人塔

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,NM,N（0<M,N<5000<M,N<500）,分别表示 A、B 的人数，保证人数合理性。

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

输入输出样例
示例
输入

1 2
输出

3

下面是Python讲师提供的解法：

import math

def check():# 判断是否可以向上拓展
    a = 0
    b = 0
    tmp = row
    while tmp >0:
        for i in range(1, tmp+1):
            if cnt[i] == 1:
                a +=1
            else:
                b += 1
        for i in range(2, tmp+1):
            if cnt[i-1]==cnt[i]:
                cnt[i-1] = 1
            else:
                cnt[i-1] = 2
        tmp -=1
    if a == m and b == n:
        return True
    else:
        return False

def dfs(k):# 遍历所有第一排
    global res
    if k > row:
        if check():
            res +=1
        return
    cnt[k] = 1
    dfs(k+1)
    cnt[k] = 2
    dfs(k+1)

if __name__ == "__main__":
    m = int(input())
    n = int(input())
    res = 0
    cnt = [0 for _ in range(100010)]
    row = int(math.sqrt(2*(m+n)))
    dfs(1)
    print(res)

原文链接：https://blog.csdn.net/qq_51718832/article/details/117265721?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522164153487816781683978391%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fblog.%2522%257D&request_id=164153487816781683978391&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~blog~first_rank_ecpm_v1~rank_v31_ecpm-1-117265721.nonecase&utm_term=%E6%9C%BA%E5%99%A8%E4%BA%BA%E5%A1%94&spm=1018.2226.3001.4450