每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0…m-1报数…这样下去…直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!_)。请你试着想下,哪个小朋友会得到这份礼品呢?(注:小朋友的编号是从0到n-1)
输入:
5,3
返回值: 3
第一种比较常规的思路是构造一个环形的链表,一直删,直到链表只有一个元素为止
# -*- coding:utf-8 -*-
class ListNode:
def __init__(self, x):
self.val = x
self.next = None
class Solution:
def LastRemaining_Solution(self, n, m):
# write code here
if n == 0 or m == 0:
return -1
if m==1 or n==1:
return n-1
head = ListNode(0)
prev = head
for i in range(1,n):
temp = ListNode(i)
prev.next = temp
prev = temp
prev.next = head
i, cnt,count = 0,0,0
temp=head
while count<n-1:
prev=temp
temp=temp.next
cnt+=1
if cnt==m-1:
prev.next = temp.next
cnt=0
count+=1
temp=prev.next
return prev.val
第二种思路是转化成类似动态规划的思路来求解
假设f(n,m)表示 n个人,报数为m,最后一个人的下表
以n=5,m=3 为例,
0,1,2,3,4 最开始编号2的人出去后,还剩4个人,初始顺序为 3,4,0,1, 3,4,0,1 可以看成是 0,1,2,3 (f(n-1,m)) 加上m 再对n 取mod 的结果。
故可以得到新的递推公式
f(n,m) =( f(n-1,m)+m) mod n
f(1,m)=0,这样就可以很容易求解拉。