第一想法是排列组合的全排解法,保证1元的数量一定比其之后的2元数量多即可,但是本题用Python作答会出现内存超限的情形
初版代码:
import itertools
M, N, K = map(int, input().split())
string = '1' * N + '2' * K
lst = list(itertools.permutations(string))
cnt = 0
if N >= K :
for data in lst :
cn1 = 0
for i in range(M) :
if data[i] == '1' :
cn1 += 1
else :
cn1 -= 1
if cn1 < 0 :
break
if cn1 >= 0 :
cnt += 1
print(cnt)
else :
print(0)
内存超限57
0.0分
0 人评分