字节跳动2018.9.20第四题

最新推荐文章于 2021-05-25 20:14:34 发布

a1310368974

最新推荐文章于 2021-05-25 20:14:34 发布

阅读量466

点赞数

CC 4.0 BY-SA版权

分类专栏： 2018秋招笔试

本文链接：https://blog.youkuaiyun.com/a1310368974/article/details/82804990

2018秋招笔试专栏收录该内容

4 篇文章

订阅专栏

本文探讨了使用动态规划算法解决n个红球、m个蓝球和k个黄球在确保相同颜色的球不相邻的情况下，计算所有可能排列组合的方法数。通过详细的代码解析，展示了如何构建状态转移矩阵并最终求得解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

n个红球，m个篮球，k个黄球，相同颜色的不相邻，多少种方法

动态规划，还行

```

num=raw_input().split(" ")
n,m,k=int(num[0]),int(num[1]),int(num[2])

result=[]
for i in range(n+1):
   tem0=[]
   for j in range(m+1):
       tem1=[]
       for p in range(k+1):
           tem2=[]
           for q in range(3):
               tem2.append(0)
           tem1.append(tem2)
       tem0.append(tem1)
   result.append(tem0)

result[1][0][0][0]=1
result[0][1][0][1]=1
result[0][0][1][2]=1
result[1][1][0][0]=1
result[1][1][0][1]=1
result[1][0][1][2]=1
result[1][0][1][0]=1
result[0][1][1][1]=1
result[0][1][1][2]=1

result[1][1][1][0]=2
result[1][1][1][1]=2
result[1][1][1][2]=2

for i in range(0,n+1):
   for j in range(0,m+1):
       for p in range(0,k+1):
           if result[i][j][p][0]==0:
               result[i][j][p][0]+=result[i-1][j][p][1]+result[i-1][j][p][2]
           if result[i][j][p][1]==0:
               result[i][j][p][1]+=result[i][j-1][p][0]+result[i][j-1][p][2]
           if result[i][j][p][2]==0:
               result[i][j][p][2]+=result[i][j][p-1][1]+result[i][j][p-1][0]

print result[n][m][k][0]+result[n][m][k][1]+result[n][m][k][2]

```