天天看点

51Nod 2738 选数字 python题解

题目链接

题目描述

从1-n中选出m个数,要求同样的数字不能重复选择,按照字典序正序输出所有方案。

例如:从1到4中选出2个数,共有6种方法,按照字典序输出,依次为:

1 2

1 3

1 4

2 3

2 4

3 4

输入样例

4 2

输出样例

1 2

1 3

1 4

2 3

2 4

3 4

思路

这道题其实就是组合问题,可以用DFS,但是这里也可以直接调用python中的itertools模块。

运用combinations(iterable, r) 创建一个迭代器,返回iterable中所有长度为r的子序列,返回的子序列中的项按输入iterable中的顺序排序

AC代码如下,非常简洁

import itertools
m,n=map(int,input().split())
ls=list(range(1,m+1))
for i in itertools.combinations(ls,n):
    print(' '.join(map(str, i)))