洛谷刷题 P1996 约瑟夫问题

洛谷刷题 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时即游戏结束。

注意

1. 当轮到下一个同学时候，index并不是进行自增操作，而是通过链表获取下一个同学的下标

index = r[index];

2. 使用链表时有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;
}