剑指62. 圆圈中最后剩下的数

问题描述

https://leetcode.cn/problems/yuan-quan-zhong-zui-hou-sheng-xia-de-shu-zi-lcof/description/

解决方案

首先,这题有更简洁的解法,但我的脑子不够用,我只会模拟。所以就只写模拟。

首先,按照解递归的一般思路,我们定义递归函数的参数和返回值。参数为li, last_idx, m。

li代表了我们需要模拟的数组,last_idx代表我们当前指针指向的值,m代表我们需要移动指针的值(含last_idx)。

返回值自然就是返回最后剩下的元素。

然后,我们来定义本层需要做什么。

由于我们要找本层需要谁出局,我们有一个计算公式:last_idx = last_idx+m-1,但这可能导致溢出,我们应当让它循环起来,即模一下li的长度。

然后我们把剩下的li交给下一层递归函数处理。

但,我们传向下一层的递归函数是什么?last_idx仔细想想,如果它不是li的最后一个元素,那么它直接指向的就是当前指针指向的“下一个”值。

如果它是li的最后一个元素,那last_idx就溢出了,我们同样要用模进行处理一下。

最后,我们设计一下递归的退出条件。

显而易见,当li数组长度为1时,只有1个幸存者,我们直接将它返回即可。

代码

class Solution:
    def lastRemaining(self, n: int, m: int) -> int:
        li = [i for i in range(n)]
        def dfs(li, last_idx, m):
            if len(li) == 1:
                return li[0]
            last_idx = (last_idx+m-1) % len(li)
            li.pop(last_idx)
            return dfs(li, last_idx%len(li), m)
        return dfs(li, 0, m)

 

posted @ 2023-01-05 12:57  BJFU-VTH  阅读(14)  评论(0编辑  收藏  举报