Python 洛谷 p1157 组合的输出

题目描述

排列与组合是常用的数学方法，其中组合就是从 n个元素中抽出 r 个元素（不分顺序且 r≤n），我们可以简单地将 n 个元素理解为自然数 1,2,…,n，从中任取 r 个数。

现要求你输出所有组合。

例如 n=5,r=3，所有组合为：

123,124,125,134,135,145,234,235,245,345。

输入格式

一行两个自然数 n,r(1<n<21,0≤r≤n)。

输出格式

所有的组合，每一个组合占一行且其中的元素按由小到大的顺序排列，每个元素占三个字符的位置，所有的组合也按字典顺序。

输出占 33 个场宽的数 xx。注意你需要头文件 iomanip。

输入输出样例

输入 #1复制

5 3 

输出 #1复制

  1  2  3
  1  2  4
  1  2  5
  1  3  4
  1  3  5
  1  4  5
  2  3  4
  2  3  5
  2  4  5
  3  4  5

from itertools import combinations

m,n=map(int,input().split())

num=[]
for i in range(m):
    num.append(i+1)

for i in combinations(num,n):
    print(''.join('%3d'%o for o in i))  #'%3d'是为了每个字符保留三个场宽

#combinations(num,n)是创建一个迭代器，返回num中所有长度为n的子序列，返回的子序列中的项按输入num中的顺序排序 (不带重复)