今天一定行


私信TA

用户名:uq_10585380137

访问量:2891

签 名:

Python苦手

等  级
排  名 770
经  验 3627
参赛次数 0
文章发表 72
年  龄 0
在职情况 学生
学  校
专  业 人工智能

  自我简介:

一位只会简单题,还妄图一行解的Noob

第一想法是排列组合的全排解法,保证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 人评分

  评论区