今天一定行


私信TA

用户名:uq_10585380137

访问量:3186

签 名:

Python苦手

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

  自我简介:

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

TA的其他文章

解题思路:排列组合类型的题目只要数据量不是很大,基本用自带的函数就行了

注意事项:注意本题可以重复取一个值,所以用itertools.combinations_with_replacement

参考代码:

import itertools

k = int(input())
for i in range(k) :
   m, n = map(int, input().split())
   lst = [i for i in range(m + 1)]
   lst_sum = []
   for data in list(itertools.combinations_with_replacement(lst, n)) :
       data = list(data)
       if sum(data) == m :
           data.sort()
           if data not in lst_sum :
               lst_sum.append(data)
   print(len(lst_sum))

 

0.0分

4 人评分

  评论区