Python解决“自助餐厅客流量分析”问题

问题描述

小A需要根据每天的客流量数据进行分析，以制定自助餐厅的备货策略。她需要计算前ii天的客流量平均值，并将其四舍五入为整数。

给定餐厅营业的总天数N，以及每天的客流量数据Ri ，你需要输出一个长度为N的序列，其中第i个值表示前i天的平均客流量。要求所有输出结果四舍五入到最接近的整数。

测试样例

样例1：
输入：N = 5 ,R = [1, 2, 3, 4, 10]
输出：[1, 2, 2, 3, 4]

样例2：
输入：N = 3 ,R = [5, 10, 15]
输出：[5, 8, 10]

解题思路

这道题目综合运用了前缀和算法知识，是一道典型的平均值计算问题。题目要求计算前i天的客流量平均值，并将其四舍五入为整数。核心信息是每天的客流量数据，需要通过前缀和的方式快速计算前i天的总客流量，然后计算平均值并进行四舍五入。通过简单的除法和取模操作来计算平均值。

1. 初始化前缀和： 使用一个变量s来存储前i天的总客流量。
2. 计算平均值： 对于每一天i，计算前i天的总客流量s，然后计算平均值s // i。为了实现四舍五入，我们需要判断余数s % i是否大于等于i/2，如果是，则平均值加1。

代码

def solution(N: int, R: list) -> list:
    # 确保输入的天数 N 和客流量数据 R 的长度一致
    assert N == len(R)
    # 初始化平均值列表
    avg = []
    # 初始化总和
    s = 0
    # 在 R 前面添加一个 0，方便后续计算
    a = [0] + R
    
    # 遍历每一天的客流量数据
    for i in range(1, N + 1):
        x = a[i]  # 当前天的客流量
        s += x  # 更新总和
        
        # 计算前 i 天的平均客流量，并四舍五入
        avg.append(s // i + ((s % i) * 2 >= i))

    # 返回平均值列表
    return avg

# 主函数，用于测试
if __name__ == '__main__':
    print(solution(N = 5 ,R = [1, 2, 3, 4, 10]))
    print(solution(N = 3 ,R = [5, 10, 15]))

输出：
[1, 2, 2, 3, 4]
[5, 8, 10]