2020春招机考汇总1（Python）：农场养鸡、连续子序列最大值的期望

最新推荐文章于 2025-01-05 00:15:00 发布

原创

最新推荐文章于 2025-01-05 00:15:00 发布 · 738 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法 #列表

本文探讨了两个有趣的算法问题：一是农场养鸡问题，通过编程模拟鸡的数量变化；二是连续子序列最大值的期望计算，涉及概率统计与序列分析。两道题目展示了算法设计与实现的技巧。

第一题：农场养鸡

n个农场，第i个农场有a[i]只鸡，每天每个农场都会增加k只鸡。每晚农场主都会选择鸡最多的农场，将该农场鸡的个数除以2下取整，在m天后剩下多少只鸡？

输入：
第一行输入三个int类型n,m,k（1 <= n,m,k <= 10^6）
第二行输入n个正整数，表示n个养鸡场初始鸡的个数
输出：
输出鸡的总数

示例：

输入：
3 3 100
100 200 400
输出：
925

代码如下：

import numpy as np
import sys


def chicken(days, nums, k):                # 编写养鸡函数
    for i in range(days):                  # 循环天数
        nums = [item+k for item in nums]   # 每天每个笼子增加k只鸡
        index = nums.index(np.max(nums))   # 找到鸡数目最大的笼子
        nums[index] = nums[index]//2       # 鸡数目减半，用整数除法
    return nums                          


if __name__ == '__main__':
    line