剑指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)