华为OD机试：一星算法题---网站统计（Python）

题目：网站统计

简单描述：

        路由器网站统计，根据输入进行网站统计，能够输出Top N的网站

输入：

        一行 如果是url 则进行统计（无需判别网站是否存在，无需输出） 如果是数字n，则按网站次数输出Top n的网站

输出：

        网站名，以逗号隔开，如果次数相同，按照url的字符字典升序排列输出

示例：

input：
qq.com
huwei.com
news.com
qq.com
2
huawei.com
news.com
huawei.com
3

output：
qq.com,huawei.com
huawei.com,news.com,qq.com

关键点：

1.输入的判断

Python中使用 isdigit() 进行判断是否为数字

2.排序

Python中使用字典，统计网站及次数 ，对字典按照次数进行排序，同时对次数相同的需要按字典升序排列

def count_website_visits():
    website_visits = {}  # 使用字典来存储网站和次数

    while True:
        line = input("请输入访问的网站或要输出的网站数量 (直接回车结束统计): ")

        if not line:
            break

        if line.isdigit():
            top_n = int(line)
            # 按照次数和URL的字符字典顺序对网站进行排序
            sorted_websites = sorted(
                website_visits.items(),
                key=lambda x: (-x[1], x[0])
            )
            top_websites = [website for website, _ in sorted_websites[:top_n]]
            top_websites_str = ", ".join(top_websites)
            print(f"排名前{top_n}的网站: {top_websites_str}")
        else:
            # 增加网站次数，如果网站不存在，则创建新的键并设置初始次数为1
            website_visits[line] = website_visits.get(line, 0) + 1


if __name__ == "__main__":
    count_website_visits()