题目

共有 n 名小伙伴一起做游戏。小伙伴们围成一圈,按 顺时针顺序 从 1 到 n 编号。确切地说,从第 i 名小伙伴顺时针移动一位会到达第 (i+1) 名小伙伴的位置,其中 1 <= i < n ,从第 n 名小伙伴顺时针移动一位会回到第 1 名小伙伴的位置。

游戏遵循如下规则:

从第 1 名小伙伴所在位置 开始 。
沿着顺时针方向数 k 名小伙伴,计数时需要 包含 起始时的那位小伙伴。逐个绕圈进行计数,一些小伙伴可能会被数过不止一次。
你数到的最后一名小伙伴需要离开圈子,并视作输掉游戏。
如果圈子中仍然有不止一名小伙伴,从刚刚输掉的小伙伴的 顺时针下一位 小伙伴 开始,回到步骤 2 继续执行。
否则,圈子中最后一名小伙伴赢得游戏。
给你参与游戏的小伙伴总数 n ,和一个整数 k ,返回游戏的获胜者。

示例 1: image.png

输入:n = 5, k = 2
输出:3
解释:游戏运行步骤如下:

  1. 从小伙伴 1 开始。
  2. 顺时针数 2 名小伙伴,也就是小伙伴 1 和 2 。
  3. 小伙伴 2 离开圈子。下一次从小伙伴 3 开始。
  4. 顺时针数 2 名小伙伴,也就是小伙伴 3 和 4 。
  5. 小伙伴 4 离开圈子。下一次从小伙伴 5 开始。
  6. 顺时针数 2 名小伙伴,也就是小伙伴 5 和 1 。
  7. 小伙伴 1 离开圈子。下一次从小伙伴 3 开始。
  8. 顺时针数 2 名小伙伴,也就是小伙伴 3 和 5 。
  9. 小伙伴 5 离开圈子。只剩下小伙伴 3 。所以小伙伴 3 是游戏的获胜者。

示例 2:

输入:n = 6, k = 5
输出:1
解释:小伙伴离开圈子的顺序:5、4、6、2、3 。小伙伴 1 是游戏的获胜者。

提示:

1 <= k <= n <= 500

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-the-winner-of-the-circular-game
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路

经典的约瑟夫环问题,下面用三种方式解,可以自己模拟、也可以使用类库中的队列、也可以使用数学法。

代码

数组模拟

  1. class Solution {
  2. public int findTheWinner(int n, int k) {
  3. // true表示淘汰
  4. boolean[] arr = new boolean[n];
  5. // 还剩 cnt 个人未淘汰
  6. int cnt = n;
  7. // 遍历下标
  8. int i = 0;
  9. while (cnt > 1) {
  10. // 要淘汰的选手
  11. int g = 0;
  12. int j = 0;
  13. // 一轮报数 k 次
  14. while (j < k) {
  15. // 未淘汰的才参与报数
  16. if (!arr[i]) {
  17. j++;
  18. // 更新要淘汰的人
  19. g = i;
  20. }
  21. i = (i + 1) % n;
  22. }
  23. arr[g] = true;
  24. cnt--;
  25. }
  26. for (int j = 0; j < n; ++j) {
  27. if (!arr[j]) return j + 1;
  28. }
  29. return -1;
  30. }
  31. }

队列模拟

  1. class Solution {
  2. public int findTheWinner(int n, int k) {
  3. Queue<Integer> q = new LinkedList<>();
  4. // 1到n依次入队
  5. for (int i = 1; i <= n; i++) {
  6. q.offer(i);
  7. }
  8. // 出队n-1个人
  9. for (int i = 1; i < n; i++) {
  10. // 前k-1个人可以在这一轮幸免,将他们移到队尾
  11. for (int j = 0; j < k - 1; j++) {
  12. q.offer(q.poll());
  13. }
  14. // 将第k个人移出去
  15. q.poll();
  16. }
  17. // 最后留下的人
  18. return q.poll();
  19. }
  20. }

数学法

公式的递推过程见「这里

  1. class Solution {
  2. public int findTheWinner(int n, int k) {
  3. int p = 0;
  4. for (int i = 2; i <= n; i++) {
  5. p = (p + k) % i;
  6. }
  7. return p + 1;
  8. }
  9. }