时间限制: 500 ms 内存限制: 65536 KB
【题目描述】
由经典约瑟夫问题改成。
有N个猴子,编号从1到N。每个猴子对应一个正整数Xi,表示如果从编号为i的猴子开始报数,需要数到Xi。
这N个猴子围成一圈,从第一个开始报数,数到第1个猴子对应的正整数X1的猴子出队,然后从它的下一位继续从1开始报数,数到对应的Xi时出队,如此循环直到剩下一个猴子,最后剩下的那个猴子就是猴子们选出的大王。
例如:
N=5,Xi对应为:1,2,3,4,5。
出队的顺序为:1,3,4,5。
【输入】
第一行为N;
第二行为N个小于等于100的正整数。对应于从某个猴子位置开始报数,需要报数的次数。
【输出】
被选为大王的猴子的编号。
【输入样例】
5
1 2 3 4 5
【输出样例】
2
【提示】
【数据范围】
N≤1000000
#include <bits/stdc++.h>
using namespace std;
void readint(int &data) {
char ch = getchar();
while (ch < '0' || ch > '9') ch = getchar();
data = 0;
do{
data = data*10 + ch-'0';
ch = getchar();
}while (ch >= '0' && ch <= '9');
}
int n, a[1000050], cnt, m;
int main()
{
cin >> n;
queue <int> q;
for(int i=1;i<=n;i++)
{
//scanf("%d",&a[i]);
readint(a[i]);
q.push(i);
}
//for(int i = 1; i <= n; i++) {
// cin >> a[i];
// q.push(i); //入队
// }
m = a[1]; // 第一个要求的数其实就是a[1]
while (q.size() != 1) {
++cnt; // 报数
int x = q.front();
q.pop();
if (cnt != m) q.push(x); // 若没到要求的数,则入队,继续报数
else { // 同循环队列
cnt = 0;
m = a[q.front()]; // 出队,并更换m
}
}
cout << q.front() << endl;
return 0;
}
/*
5
1 2 3 4 5
*/