游戏排名第三大的分数 | 豆包MarsCode AI刷题

42 阅读1分钟

def solution(n: int, nums: list) -> int:
    # 使用集合去重
    unique_scores = set(nums)
    
    # 将集合转换为列表并排序
    sorted_scores = sorted(unique_scores, reverse=True)
    
    # 判断列表长度并返回结果
    if len(sorted_scores) >= 3:
        return sorted_scores[2]  # 返回第三大的分数
    else:
        return sorted_scores[0]  # 返回最大的分数

if __name__ == '__main__':
    print(solution(3, [3, 2, 1]) == 1)
    print(solution(2, [1, 2]) == 2)
    print(solution(4, [2, 2, 3, 1]) == 1)