题目描述
n 个人的编号是 1 ~ n,如果他们依编号按顺时针排成一个圆圈,从编号是1的人开始顺时针报数。
(报数是从 1 报起)当报到 k 的时候,这个人就退出游戏圈,下一个人重新从 1 开始报数。
求最后剩下的人的编号。这就是著名的约瑟夫环问题。
本题目就是已知 n,k 的情况下,求最后剩下的人的编号。
输入格式
题目的输入是一行,2 个空格分开的整数 n, k
输出格式
要求输出一个整数,表示最后剩下的人的编号。
样例输入
10 3
样例输出
4
数据范围
0 < n, k < 105
题解一
队列(会超时):
#include <iostream>
#include <queue>
using namespace std;
queue<int> q;
int main()
{
int n, k;
cin >> n >> k;
for (int i = 1; i <= n; i ++) q.push(i);
int num = 1;
while(q.size()!= 1)
{
if(num == k)
{
q.pop();
num = 1; // 重新报数
}
else
{
q.push(q.front()); // 将队头放到队尾
q.pop();
num ++;
}
}
cout << q.front() << endl;
return 0;
}
题解二
递推:
f[i]
:有 i
个人参加游戏时,剩下的人的编号(下标从 0 开始)。
由于下标从 0 开始,所以报 k
的是第 k-1
个人。从下一个人开始,所有人重新从 0 开始编号,比如第 k
个人的编号变为 0
,…
那么问题就转变成:有 i-1
个人参加游戏,剩下的人的编号,即 f[i - 1]
。
原编号与新编号相差 k
,所以f[i] = f[i - 1] + k
,又因为编号不能超过人数,所以还要 % i
。
#include <iostream>
using namespace std;
int f[1000010];
int main()
{
int n, k;
cin >> n >> k;
for (int i = 1; i <= n; i ++) f[i] = (f[i - 1] + k) % i;
cout << f[n] + 1 << endl;
return 0;
}