python怎么涂色_涂色问题（Python）

题目：将一个圆形等分成N个小扇形，将这些扇形标记为1，2，3,…，N。现在使用M种颜色对每个扇形进行涂色，每个扇形涂一种颜色，且相邻的扇形颜色不同,问有多少种不同的涂法?(N≥1，M≥3)

递归解决：

当n=1是，f(1,m) = m

当n=2是，f(1,m) = m(m-1)

当n=3是，f(1,m) = m(m-1)(m-2)

当n=4时，f(4,m) = m(m-1)(m-2)(m-2) + m(m-1)(m-1) = m(m-1)(m^2-3m+3)

当n=5时，f(5,m) = m(m-1)(m-2)(m-2) + m(m-1)(m-1)(m-1)= m(m-1)(m^2-3m+3)(m-1)= f(4,m)(m-1)

可推出关系 f(n,m) = f(n-1,m)(m-1)

def calculate(n,m):

"""

:param n: n个扇形

:param m: m种颜色

:return: 涂法的种类

"""

if n == 1:

return m

if n == 2:

return m * (m-1)

if n == 3:

return m * (m-1) * (m-2)

if n == 4:

return m * (m-1) * (m-2) * (m-2) + m * (m-1) * (m-1)

return calculate(n-1,m) * (m-1)

print(calculate(4,4))