python解决约瑟夫环问题

Python解决约瑟夫环问题

题目：一排人，编号从1开始到17。所有人按序号围成一个圈，从编号1开始，挨个报数，从1开始，报到3则出局.然后再次从1开始报数，直到只剩下一个人。问最后那个人编号为几。

我个人来说嘴比较笨，不过代码和想法都是自己的。我就直接上代码了

def getnum():#赋值1,2,3
    i=1
    while True:
        yield i
        i=i+1
        if i ==4:
            i=1
a=getnum()#迭代器
list1=[i for i in range(1,18)]#17个人
list2=[i for i in range(1,len(list1)+1)]
list3=dict([i for i in zip(list1,list2)])#先构造字典
while len(list3)>1:
    for i in list3.keys():
        list3[i]=next(a)#每轮每个人报的数字
    print(list3)
    keys=list(list3.keys())
    for key in keys: #剔除报的数字是3的人
        if list3.get(key)==3:
            del list3[key]
print('最后留下的人的编号为%d'%list(list3.keys())[0])

运行截图如下：

希望大佬能给指正一下，或者有更好的想法可以相互交流一下。本人菜鸟一枚。