华为od机试 Python【拔河比赛】

墨蓝 2024-03-08 06:26 65阅读 0赞

### 题目 ###

我们需要为拔河比赛挑选人选，挑选规则如下:  
1首先按身高排序，然后按体重排序.  
2.选出10个最合适的人选

输入是一个数组，数组存储的是所有人员的身高、体重信息，【身高,体重】;部门的总人数大于10。

输入描述  
输入为N行员工信息

如  
181 70  
182 70

这表示第一人身高181厘米，体重70公斤;第二人身高182厘米，体重70公斤

示例1：

输入  
181 70  
182 70  
183 70  
184 70  
185 70  
186 70  
180 71  
180 72  
180 73  
180 74  
180 75

输出  
186 70  
185 70  
184 70  
183 70  
182 70  
181 70  
180 75  
180 74  
180 73  
180 72

### 代码 ###

class Person:
        def __init__(self, height, weight):
            self.height = height
            self.weight = weight
    
        def __repr__(self):
            return f"{
          self.height} {
          self.weight}"
    
    # 选出拔河比赛的前10名
    def select_top_10(people):
        # 先按身高降序，若身高相同则按体重降序
        people.sort(key=lambda p: (-p.height, -p.weight))
        return people[:10]
    
    def main():
        # 题目给出的测试case
        people = [
            Person(181, 70),
            Person(182, 70),
            Person(183, 70),
            Person(184, 70),
            Person(185, 70),
            Person(186, 70),
            Person(180, 71),
            Person(180, 72),
            Person(180, 73),
            Person(180, 74),
            Person(180, 75)
        ]
    
        for person in select_top_10(people):
            print(person)
    
    if __name__ == "__main__":
        main()