阿里2020.3.30笔试题——养鸡场

养鸡场鸡群增长模拟

原创于 2021-03-15 16:51:24 发布 · 392 阅读

0 ·

CC 4.0 BY-SA版权

编程同时被 3 个专栏收录

34 篇文章

订阅专栏

算法

29 篇文章

订阅专栏

实习

13 篇文章

订阅专栏

本文介绍了一种使用大顶堆数据结构解决养鸡场鸡群增长问题的方法。问题设定为：给定初始鸡场数量及养鸡天数，每天开始时各鸡场增加固定数量的鸡，每天结束时数量最多鸡场的鸡减半。文章通过示例代码展示了如何高效地求解最终所有鸡场的鸡总数。

问题描述

给定鸡场数量 $n$ ，养鸡天数 $m$ ，每次增加数量 $k$
每一天开始时，每个养鸡场都增加k只鸡；每一天结束时，将数量最多的鸡场数量减半（向下取整）

问题分析

因为每次选择数量最多的鸡场，所以使用大顶堆即可。

代码示例

import heapq as hp

def solve():
    N, M, K = map(int, input().split())
    chicken_list = list(map(int, input().split()))
    chicken_list = list(map(lambda x: -x, chicken_list))
    hp.heapify(chicken_list)
    for i in range(M):
        tmp = -hp.heappop(chicken_list) - (i + 1) * K
        tmp = tmp // 2
        hp.heappush(chicken_list, -tmp)
    
    res = 0
    for i in range(N):
        res += -hp.heappop(chicken_list)
    return res + N * M * K

print(solve())