【每日一题】面试题62 画圈中最后剩下的数字（约瑟夫问题）

约瑟夫问题

题目竟然还是简单的，但是受限于复杂度，没办法用常规方法解答。因此引入了一个具有dp思想的约瑟夫问题。
具体的解法是一个数学问题，我就直接放图了。
思想是将大问题化为了子问题，从底至上的解决。

class Solution(object):
    def lastRemaining(self, n, m):
        """
        :type n: int
        :type m: int
        :rtype: int
        """
        res = 0
        for i in range(2, n + 1):
            res = (res + m) % i
        return res

例题2 升级版

这次每次迭代时候，删除的位置时不同的了。我们还是需要明确思路。假设每次删除的是第m号，此时剩余的个数为i
最后一次开始时，一定是从index = 0开始。
因此，在倒数第二次开始时，它的位置是index = (0+m)%2
因此，在倒数第三次开始时，它的位置是(index + m)%3
…
可见，是倒序的递推。

T = int(input())

def helper(nums, m):
    n = len(nums)
    i = 2
    index = 0
    for i in range(2,m+1):
    # 这里是一次找到，倒数第二次的m，倒数第三的m
        index = (index+int(nums[(m-i)%n])) % i
    return index

for i in range(T):
    m,n,*nums = map(int, input().split())
    index = helper(nums, m)
    print(index)