Python约瑟夫问题

题目内容：

已知n个人（以编号0，1，2，3...n-1分别表示）围坐在一张圆桌周围。从编号为0的人开始报数1，数到m的那个人出列；他的下一个人又从1开始报数，数到m的那个人又出列；依此规律重复下去，直到圆桌周围的人全部出列。

 

输入格式:

两个正整数n, m，其中3<=n<=100, 1<=m<=n

 

输出格式：

按照顺序出列的人的编号列表

 

输入样例：

5

2

 

输出样例：

[1, 3, 0, 4, 2]

 

输入样例：

12

3

 

输出样例：

[2, 5, 8, 11, 3, 7, 0, 6, 1, 10, 4, 9]

 

编号从1到N，如果退出，则将编号变为0。循环n次退出

def josephus(n,m):
    """
        环的问题,
        共有n个人围成一圈,从1开始报数,数到m时退出,再从1开始,直到所有人退出
    """
    people = list(range(1,n+1))
    index = 0
    res = []
    #给n个人编号放到表people中,从下标为0的人开始
    for i in range(n):
        #for循环用来控制内部循环执行次数
        count = 0
        while count < m:
            if people[index] != 0:
                count += 1
            if count == m:
                res.append(people[index]-1)
                #把退出的人的编号置0
                people[index] = 0
            index = (index+1) % n
    return res

if __name__ == '__main__':
    n = int(input())
    m = int(input())
    res = josephus(n,m)
    print(res)