Python入门习题（23）——CCF CSP认证考试真题：卖菜

问题描述

试题编号： 201809-1
试题名称： 卖菜
时间限制： 1.0s
内存限制： 256.0MB

问题描述
　　在一条街上有n个卖菜的商店，按1至n的顺序排成一排，这些商店都卖一种蔬菜。
　　第一天，每个商店都自己定了一个价格。店主们希望自己的菜价和其他商店的一致，第二天，每一家商店都会根据他自己和相邻商店的价格调整自己的价格。具体的，每家商店都会将第二天的菜价设置为自己和相邻商店第一天菜价的平均值（用去尾法取整）。
　　注意，编号为1的商店只有一个相邻的商店2，编号为n的商店只有一个相邻的商店n-1，其他编号为i的商店有两个相邻的商店i-1和i+1。
　　给定第一天各个商店的菜价，请计算第二天每个商店的菜价。
输入格式
　　输入的第一行包含一个整数n，表示商店的数量。
　　第二行包含n个整数，依次表示每个商店第一天的菜价。
输出格式
　　输出一行，包含n个正整数，依次表示每个商店第二天的菜价。
样例输入
8
4 1 3 1 6 5 17 9
样例输出
2 2 1 3 4 9 10 13
数据规模和约定
　　对于所有评测用例，2 ≤ n ≤ 1000，第一天每个商店的菜价为不超过10000的正整数。

解题思路

1. 计算1号商铺第二天的菜价；计算n号商铺第二天的菜价。这两个商铺只有1个邻居。
2. 计算2, 3, …, n-1号商铺的第二天的菜价。这些商铺有2个邻居。

参考答案

n = int(input())
prices = [int(s) for s in input().split()]

prices_next_day = [0] * len(prices)
prices_next_day[0] = (prices[0] + prices[1]) // 2  #1号商铺第二天的菜价
prices_next_day[-1] = (prices[-1] + prices[-2]) // 2  #n号商铺第二天的菜价

#2, 3, 4, ..., n-1号商铺第二天的菜价
for i in range(1, len(prices) - 1):
    prices_next_day[i] = (prices[i - 1] + prices[i] + prices[i + 1]) // 3

for p in prices_next_day:
    print(p, end=' ')

测试用例

测试用例用来验证程序代码的正确性。题目描述中给出的测试用例（见样例输入和样例输出）是一般情形。菜价数据已经覆盖了恰好整除的情形（(2+2)/2 = 2；(2+1+3)/3=2）和没有整除的情形（(4+1)/2=2；(4+1+3)/3=2）。没有整除的情形会进行去尾法取整。
　　下面补充覆盖其他情形的测试用例。

1. n=2的情形。
   样例输入
   2
   3 4
   样例输出
   3 3

注：如果你发现上述测试用例没有覆盖到的情形，请在评论中指出。

小结

1. 先计算1号和n号商铺第二天菜价，再计算2, 3, …, n-1号商铺第二天的菜价。这样的分离使得计算2, 3, …, n-1号商铺第二天的菜价的循环变得简短又清晰。