2018.9.20字节跳动第四场研发岗在线笔试编程题第4题

有r个红球,y个黄球,b个蓝球,同色的球完全相同,问排成一列且同色球不相邻的方式有几种?

设函数f,初始条件f(0,0,0)=0,f(1,0,0)=f(0,1,0)=f(0,0,1)=1,f(1,2,0)=1,f(1,1,0)=f(2,2,0)=2

当只排列两色球且同色球不相邻时,可得

类似地考虑加入第三种颜色的球,最终递归公式为:

Case:对于函数f(i,j,k),满足 i+j>=k-1,i+k>=j-1,且 j+k>=i-1

解释:

r红y黄b蓝的排列个数,等于假设以红色开始,(r-1)红y黄b蓝的个数;加上以黄色,以蓝色开始

但是这个(r-1)红y黄b蓝本身不能以红色开头,所以得减去以红色开头(r-2)红y黄b蓝的情况

这个(r-2)红y黄b蓝本身也需要排除红色开头,所以自己得减去(r-3)红y黄b蓝的情况。。以此类推

 

 

 

你可能感兴趣的:(AI笔试)