算法题目 (Python版本)

1. n个扇形m个颜色着色相邻扇形颜色不相同问题（Python）

import numpy as np

def colornum(n, m):
    if n == 0:
        return 0
    elif n == 1:
        return m
    elif n == 2:
        return m * (m-1)

    else:
        return m * np.power(m-1, n-1) - colornum(n-1, m)

print(colornum(3,2))