软工日报5.13

8-1 【Python0002】排列组合序列

分数 10

作者 doublebest

单位石家庄铁道大学

【题目描述】用户输入整数n（1<=n<=26）和整数m（m<=n），然后输入n个不同的字母，请编写程序输出在这n个字母中选择m个字母的所有排列序列和组合序列。
【练习要求】请给出源代码程序和运行测试结果，源代码程序要求添加必要的注释。
【输入格式】在第一行中输入整数n和整数m的值，数据之间以空格为间隔。
在第二行中输入n个不同字母，数据之间以空格为间隔
【输出格式】首先输出所有的排列数列，每种情况一行，字母间以空格为间隔；
再者首先输出所有的组合数列，每种情况一行，在同一行中以字母增序顺序显示，字母间以空格为间隔。
【输入样例】3 2
a s d
【输出样例】Permutation:
a d
d a
a s
s a
d s
s d
Combination:
a d
a s
d s

代码：

from itertools import permutations, combinations

def get_input():
    # 获取用户输入
    n = int(input("请输入整数n（1<=n<=26）："))
    m = int(input("请输入整数m（m<=n）："))

    # 检查输入的有效性
    if n < 1 or n > 26 or m > n:
        print("输入不合法，请确保1<=n<=26且m<=n。")
        return None, None

    # 输入n个不同的字母
    print(f"请输入{n}个不同的字母，用空格分隔：")
    letters = input().split()

    # 检查字母数量和唯一性
    if len(letters) != n or len(set(letters)) != n:
        print("输入的字母数量不正确或有重复，请重新输入。")
        return None, None

    return n, m, letters

def display_sequences(n, m, letters):
    # 生成并打印所有排列
    print("\n所有排列序列：")
    for p in permutations(letters, m):
        print(''.join(p))

    # 生成并打印所有组合
    print("\n所有组合序列：")
    for c in combinations(letters, m):
        print(''.join(c))

if __name__ == "__main__":
    n, m, letters = get_input()
    if n and m and letters:
        display_sequences(n, m, letters)