📌  相关文章
📜  打印所有小于N的数字,最多2个唯一数字(1)

📅  最后修改于: 2023-12-03 15:10:02.138000             🧑  作者: Mango

打印所有小于N的数字,最多2个唯一数字

在编程中,有时候需要打印所有小于某个数字N的数字,但是又只需要打印最多2个唯一数字。这个需求在实际的开发中很常见,比如打印某个范围内的不同年龄段用户数量。

下面给出一个Python代码示例,它能够实现这个需求。该程序首先生成一个包含所有小于N的数字的列表,然后使用set()函数去重,最后判断去重后的数字个数是否不超过2,如果不超过2则全部打印出来,否则只打印前2个数字。

def print_numbers_less_than_n_with_2_distinct_elements(n):
    numbers = list(range(n))
    distinct_numbers = set(numbers)
    if len(distinct_numbers) <= 2:
        print(distinct_numbers)
    else:
        print(set(list(distinct_numbers)[:2]))

例子:当n等于10时,程序输出:

{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}

例子:当n等于5时,程序输出:

{0, 1, 2, 3, 4}

例子:当n等于2时,程序输出:

{0, 1}

此代码可适用于大部分Python版本。