【华为od机试】约瑟夫问题-Python3

■ 题目描述

【约瑟夫问题】

输入一个由随机数组成的数列（数列中每个数均是大于 0 的整数，长度已知），和初始计数值 m。

从数列首位置开始计数，计数到 m 后，将数列该位置数值替换计数值 m，

并将数列该位置数值出列，然后从下一位置从新开始计数，直到数列所有数值出列为止。

如果计数到达数列尾段，则返回数列首位置继续计数。

请编程实现上述计数过程，同时输出数值出列的顺序。

比如：输入的随机数列为：3,1,2,4，初始计数值 m=7，从数列首位置开始计数（数值 3 所在位置）

第一轮计数出列数字为 2，计数值更新 m=2，出列后数列为 3,1,4，从数值 4 所在位置从新开始计数

第二轮计数出列数字为 3，计数值更新 m=3，出列后数列为 1,4，从数值 1 所在位置开始计数

第三轮计数出列数字为 1，计数值更新 m=1，出列后数列为 4，从数值 4 所在位置开始计数

最后一轮计数出列数字为 4，计数过程完成。输出数值出列顺序为：2,3,1,4。

要求实现函数：

void array_iterate(int len, int input_array[], int m, int output_array[])

【输入】 int len：输入数列的长度； int intput_array[]：输入的初始数列 int m：初始计数值

【输出】 int output_array[]：输出的数值出列顺序

【返回】 无示例:

输入： int input_array[] = {3,1,2,4}，int len = 4， m=7

输出： output_array[] = {2,3,1,4}

def array_iterate(n : int , input_array:list, m:int) -> list :
    output_array = [] # 输出数组
    choose = 0 #被选中准备剔除的下标
    while n > 0:
        start = choose
        choose = (start+(m-1)%n)%n
        print("m:",m,"choose :",choose,"start:",start,input_array)
        m = input_array.pop(choose)
        output_array.append(m)
        n -= 1
    return output_array


if __name__ == '__main__':
    lens = 4
    m = 7
    input_array = [3,1,2,4]
    res = array_iterate(lens,input_array,m)
    print(res)