python版約瑟夫環

#! /usr/bin/env python3
#coding = utf-8

import itertools


def josef(list0, start, interval):  # start爲開始刪除的元素下標，interval爲間隔
    removed_list = []  # 創建用於保存被刪除元素的列表
    round_cycle = itertools.cycle(list0)  # 創建循環遍歷list0的生成器
    for i in range(start):
        current_element = next(round_cycle)  # 定位當前元素
    removed_list.append(current_element)  # 將start位置的元素放入刪除列表
    while len(removed_list) < len(list0):
        count = 0  # 計數放在for循環外，每剔除一個元素後歸零
        for i in range(interval):  # 循環開始
            while count < interval:  # 計數開始
                current_element = next(round_cycle)
                if current_element not in removed_list:
                    count += 1
        removed_list.append(current_element)
    print('removed_list={}'.format(removed_list))
    return removed_list[-1]


def main():
    test_list = [i for i in range(1, 10)]
    print('test_list={}'.format(test_list))
    print('last_one={}'.format(josef(list0=test_list, start=2, interval=3)))

if __name__ == '__main__':
    main()

# 結果輸出
'''
test_list=[1, 2, 3, 4, 5, 6, 7, 8, 9]
removed_list=[2, 5, 8, 3, 7, 4, 1, 6, 9]
last_one=9
'''