python中的着色网格

Question

系统会为您提供1×n网格和k颜色。您的任务是通过遵守以下条件来确定对每个单元进行着色的方式： 对于将网格划分为两个非空部分的任何垂直线，右侧的不同颜色的数量必须等于左侧的不同颜色的数量

输入格式： 输入的第一行包含两个整数n和k

输出格式： 按照上述条件打印为每个单元着色的方法。

SAMPLE INPUT 10 5

SAMPLE OUTPUT   5

我的代码：

import math    

def color():
    n, r = map(int, input().split())
    n = n/2
    return int(math.factorial(n)/(math.factorial(n-r)*math.factorial(r)))  

print(color())