leetcode 2682. 找出转圈游戏输家

205 阅读1分钟

n 个朋友在玩游戏。这些朋友坐成一个圈,按 顺时针方向 从 1 到 n 编号。从第 i 个朋友的位置开始顺时针移动 1 步会到达第 (i + 1) 个朋友的位置(1 <= i < n),而从第 n 个朋友的位置开始顺时针移动 1 步会回到第 1 个朋友的位置。

游戏规则如下:

第 1 个朋友接球。

  • 接着,第 1 个朋友将球传给距离他顺时针方向 k 步的朋友。
  • 然后,接球的朋友应该把球传给距离他顺时针方向 2 * k 步的朋友。
  • 接着,接球的朋友应该把球传给距离他顺时针方向 3 * k 步的朋友,以此类推。

换句话说,在第 i 轮中持有球的那位朋友需要将球传递给距离他顺时针方向 i * k 步的朋友。

当某个朋友第 2 次接到球时,游戏结束。

在整场游戏中没有接到过球的朋友是 输家 。

给你参与游戏的朋友数量 n 和一个整数 k ,请按升序排列返回包含所有输家编号的数组 answer 作为答案。

示例 1:

输入: n = 5, k = 2
输出: [4,5]
解释: 以下为游戏进行情况:
1)第 1 个朋友接球,第 1 个朋友将球传给距离他顺时针方向 2 步的玩家 —— 第 3 个朋友。
2)第 3 个朋友将球传给距离他顺时针方向 4 步的玩家 —— 第 2 个朋友。
3)第 2 个朋友将球传给距离他顺时针方向 6 步的玩家 —— 第 3 个朋友。
4)第 3 个朋友接到两次球,游戏结束。

示例 2:

输入: n = 4, k = 4
输出: [2,3,4]
解释: 以下为游戏进行情况:
1)第 1 个朋友接球,第 1 个朋友将球传给距离他顺时针方向 4 步的玩家 —— 第 1 个朋友。
2)第 1 个朋友接到两次球,游戏结束。

提示:

  • 1 <= k <= n <= 50

题解分析: n代表人数,k代表每次累加数量,i代表第几次接球,index为起始位,那么每个接球的操作抽象出来就是index的重新计算:

  • 新地址为 newIndex = (index + ik) % n == 0 ? n : (index + ik) % n
  • 最多是执行n次。

方法一: 设置一个Set保存未出现的位置(默认没有第一个位置),每次计算位置后,判断是否在Set中有值,若没有即游戏结束,返回Set中的值,代码如下:

public int[] circularGameLosers(int n, int k) {
    Set<Integer> set = new HashSet<>();
    for (int i = 2; i <= n; i++) {
        set.add(i);
    }
    int currentIdx = 1;
    int i = 1;
    while (true) {
        if (!set.contains(currentIdx) && i != 1) {
            break;
        } else {
            set.remove(currentIdx);
            int temp = (currentIdx + i * k);
            if(temp % n == 0){
                currentIdx = n;
            }else {
                currentIdx = temp%n;
            }
            i++;
        }

    }
    int[] result = new int[set.size()];
    int index = 0;
    for (int num : set) {
        result[index++] = num;
    }
    return result;
}