本文涉及知识点
数学 约瑟夫环
LeetCode1823. 找出游戏的获胜者
共有 n 名小伙伴一起做游戏。小伙伴们围成一圈,按 顺时针顺序 从 1 到 n 编号。确切地说,从第 i 名小伙伴顺时针移动一位会到达第 (i+1) 名小伙伴的位置,其中 1 <= i < n ,从第 n 名小伙伴顺时针移动一位会回到第 1 名小伙伴的位置。
游戏遵循如下规则:
从第 1 名小伙伴所在位置 开始 。
沿着顺时针方向数 k 名小伙伴,计数时需要 包含 起始时的那位小伙伴。逐个绕圈进行计数,一些小伙伴可能会被数过不止一次。
你数到的最后一名小伙伴需要离开圈子,并视作输掉游戏。
如果圈子中仍然有不止一名小伙伴,从刚刚输掉的小伙伴的 顺时针下一位 小伙伴 开始,回到步骤 2 继续执行。
否则,圈子中最后一名小伙伴赢得游戏。
给你参与游戏的小伙伴总数 n ,和一个整数 k ,返回游戏的获胜者。
示例 1:
输入:n = 5, k = 2
输出:3
解释:游戏运行步骤如下:
- 从小伙伴 1 开始。
- 顺时针数 2 名小伙伴,也就是小伙伴 1 和 2 。
- 小伙伴 2 离开圈子。下一次从小伙伴 3 开始。
- 顺时针数 2 名小伙伴,也就是小伙伴 3 和 4 。
- 小伙伴 4 离开圈子。下一次从小伙伴 5 开始。
- 顺时针数 2 名小伙伴,也就是小伙伴 5 和 1 。
- 小伙伴 1 离开圈子。下一次从小伙伴 3 开始。
- 顺时针数 2 名小伙伴,也就是小伙伴 3 和 5 。
- 小伙伴 5 离开圈子。只剩下小伙伴 3 。所以小伙伴 3 是游戏的获胜者。
示例 2:
输入:n = 6, k = 5
输出:1
解释:小伙伴离开圈子的顺序:5、4、6、2、3 。小伙伴 1 是游戏的获胜者。
提示:
1 <= k <= n <= 500
约瑟夫环
i ∈ \in ∈[0,n-1]。next(i) = (i+1)%n。删除i后,next(i-1)=(i+1)% 。删除i后,仍然是约瑟夫环。
i+1…n-1 变成 0…m ,0…i-1变成m+1…n-2。
以原始编号0为起点的长度为n环,变成以原始编号i+1为起点,长度为n-1的环。
f(n):以原始编号0为起点,长度为n的环,最后获胜者的编号。
n =1 ,f(n)返回0。
n >1,( k + F(n - 1))%n
最终结果:f(n)%n+1
代码
核心代码
class Solution {
public:
int findTheWinner(int n, int k) {
function<int(int)> F = [&](int n) {
if (1 == n) { return 0; }
return ( k + F(n - 1))%n;
};
const int ans = F(n);
return ans +1;
}
};
单元测试
TEST_METHOD(TestMethod1)
{
auto res = Solution().findTheWinner(5, 1);
AssertEx(5, res);
}
TEST_METHOD(TestMethod2)
{
auto res = Solution().findTheWinner(3, 2);
AssertEx(3, res);
}
TEST_METHOD(TestMethod3)
{
auto res = Solution().findTheWinner(2, 2);
AssertEx(1, res);
}
TEST_METHOD(TestMethod4)
{
auto res = Solution().findTheWinner(4, 2);
AssertEx(1, res);
}
TEST_METHOD(TestMethod11)
{
auto res = Solution().findTheWinner(5, 2);
AssertEx(3, res);
}
TEST_METHOD(TestMethod12)
{
auto res = Solution().findTheWinner(6, 5);
AssertEx(1, res);
}