每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0...m-1报数....这样下去....直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!^_^)。请你试着想下,哪个小朋友会得到这份礼品呢?(注:小朋友的编号是从0到n-1)
1 # -*- coding:utf-8 -*- 2 class Solution: 3 def LastRemaining_Solution(self, n, m): 4 if n == 0: 5 return -1 6 if n == 1: 7 return 0 8 dp = [0] * (n+1) 9 dp[0] = -1 10 dp[1] = 0 11 for i in range(2,n+1): 12 dp[i] = (dp[i-1] + m) % i 13 print(dp) 14 return dp[n] 15 # write code here
Java版代码,leetcode地址:
1 public int lastRemaining(int n, int m) { 2 if (n == 0) { 3 return -1; 4 } 5 if (n == 1) { 6 return 0; 7 } 8 int[] dp = new int[n + 1]; 9 dp[0] = -1; 10 dp[1] = 0; 11 for (int i = 2; i < n + 1; i++) { 12 dp[i] = (dp[i - 1] + m) % i; 13 } 14 return dp[n]; 15 }