蓝桥杯--分糖果

蓝桥杯–01分糖果

问题描述

有n个小朋友围坐成一圈。老师给每个小朋友随机发偶数个糖果，然后进行下面的游戏：每个小朋友都把自己的糖果分一半给左手边的孩子。一轮分糖后，拥有奇数颗糖的孩子由老师补给1个糖果，从而变成偶数。反复进行这个游戏，直到所有小朋友的糖果数都相同为止。你的任务是预测在已知的初始糖果情形下，老师一共需要补发多少个糖果。
输入格式
程序首先读入一个整数N(2<N<100)，表示小朋友的人数。接着是一行用空格分开的N个偶数（每个偶数不大于1000，不小于2）
输出格式
要求程序输出一个整数，表示老师需要补发的糖果数。
样例输入
3 2 2 4
样例输出
4

思路解析

模拟整个过程即可，注意边界值(result[0]的更新)的处理

def tansmit_candy(result):
    """
    分一半给左手边
    :param result: 各小孩拥有糖果数
    :return: 新的拥有糖果情况
    """
    start = result[0]
    end = result[len(result) - 1]
    for i in range(1, len(result)):
        result[i - 1] += result[i] // 2
        result[i] /= 2
    result[len(result) - 1] += start // 2
    result[0] -= start // 2


def supply(result):
    """
    补发糖果给拥有奇数个糖果的小孩，注意边界值的处理
    :param result:
    :return:
    """
    global supply_number
    for i in range(len(result)):
        if result[i] % 2:
            result[i] += 1
            supply_number += 1


def game_end(result):
    """
    判断游戏是否结束，即所有小孩糖果数是否相同
    :param result:
    :return:
    """
    for i in range(len(result)):
        if result[i] != result[0]:
            return False
    return True


if __name__ == '__main__':
    N = int(input())
    result = []
    supply_number = 0
    for i in range(N):
        temp = int(input())
        result.append(temp)
    while True:
        if not game_end(result):
            tansmit_candy(result)
            supply(result)
        else:
            break
    print(supply_number)