洛谷刷题 P1996 约瑟夫问题
问题如下
n 个人围成一圈,从第一个人开始报数,数到 m 的人出列,再由下一个人重新从 1 开始报数,数到 m 的人再出圈,依次类推,直到所有的人都出圈,请输出依次出圈人的编号。
输入格式
输入两个整数 n,m (1 < n,m < 100)
输出格式
输出一行 n 个整数,按顺序输出每个出圈人的编号。
思路
我使用两个数组l[N],r[N]储存了一个双向链表,指向第i个人左侧和右侧的对象,从而链接每个人。
开始我用下标为1的元素储存第一个同学,用下标为n的元素储存第n个同学,然后用循环将剩下n-2个同学依次加入。
同时我设置了now 和 index两个变量存储报数的数字和当前位置,当报数的数字=m时,输出index并且将下标为index的元素从双向链表中移除,n–。当n=0时即游戏结束。
注意
- 当轮到下一个同学时候,index并不是进行自增操作,而是通过链表获取下一个同学的下标
index = r[index];
- 使用链表时有struct 结构体和数组两种表示方法,但是使用结构体会导致new速度较慢,所以采用数组存储双向链表。
#include<iostream>
#include<string.h>
#include<string>
#include<iomanip>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 250;
//l[i] i下标左边的点
int e[N], l[N], r[N];
void remove(int k) {
r[l[k]] = r[k];
l[r[k]] = l[k];
}
int n, m;
void init() {
//1表示左端点
l[1] = n, r[1] = 2;
l[n] = n-1, r[n] = 1;
for (int i = 2; i <= n - 1; i++) {
l[i] = i - 1, r[i] = i + 1;
}
}
int main() {
init();
cin >> n >> m;
int now = 1;
int index = 1;
while (n > 0) {
if (now == m) {
cout << index << " ";
remove(index),n--;
now = 1;
}
else {
now++;
}
index = r[index];
}
return 0;
}