小扣注意到秋日市集上有一个创作黑白方格画的摊位。摊主给每个顾客提供一个固定在墙上的白色画板,画板不能转动。画板上有 n * n
的网格。绘画规则为,小扣可以选择任意多行以及任意多列的格子涂成黑色(选择的整行、整列均需涂成黑色),所选行数、列数均可为 0。
小扣希望最终的成品上需要有 k
个黑色格子,请返回小扣共有多少种涂色方案。
注意:两个方案中任意一个相同位置的格子颜色不同,就视为不同的方案。
class Solution:
def paintingPlan(self, n: int, k: int) -> int:
# 无放回排列组合函数
def SortNum( i: int, j: int):
res = 1
while i:
res = res * j / i
j = j - 1
i = i -1
return res
if k == 0 :
return 1
if k == n * n:
return 1
if k < n:
return 0
count = 0
for i in range(n+1) :
for j in range(n+1) :
if i*n + j*(n-i) == k:
count = count + SortNum(i,n) * SortNum(j,n)
print(i,j,count)
return int(count)