约瑟夫环

# -*- coding: utf-8 -*-
# @Time         : 2019-09-18 21:57
# @Author       : Jayce Wong
# @ProjectName  : job
# @FileName     : josephus.py
# @Blog         : https://blog.51cto.com/jayce1111
# @Github       : https://github.com/SysuJayce

"""
约瑟夫斯（Josephus）问题是一个出如今计算机科学和数学中的问题。
在计算机编程的算法中，相似问题又称为约瑟夫环。
约瑟夫斯问题：有n个囚犯站成一个圆圈，准备处决。
首先从一我的开始，越过k-2我的（由于第一我的已经被越过），并杀掉第k我的。
接着，再越过k-1我的，并杀掉第k我的。
这个过程沿着圆圈一直进行，直到最终只剩下一我的留下，这我的就能够继续活着。

给定了n和k，一开始要站在什么地方才能避免被处决？

递推公式：
当n = 1时，f(1, k) = 1
当n > 1时，f(n, k) = (f(n - 1, k) + k) mod n
**注意**当编号从1开始的时候，若是计算获得f(n, k) = 0，那么须要将其还原为n而后继续递推
"""

def josephus(n, k):
    if n <= 1:
        return 1
    res = 1
    # 注意这里咱们使用递推公式的时候，计算的是f(i, k)，所以须要对i取模
    for i in range(2, n + 1):
        res = (res + k) % i if (res + k) % i != 0 else i
    return res

print(josephus(5, 2))