【洛谷】P1996 约瑟夫问题

题目地址：

https://www.luogu.com.cn/problem/P1996

题目描述：
$n$个人围成一圈，从第一个人开始报数,数到$m$的人出列，再由下一个人重新从$1$开始报数，数到$m$的人再出圈，依次类推，直到所有的人都出圈，请输出依次出圈人的编号。

注意：本题和《深入浅出-基础篇》上例题的表述稍有不同。书上表述是给出淘汰$n-1$名小朋友，而该题是全部出圈。

输入格式：
输入两个整数$n,m$。

输出格式：
输出一行$n$个整数，按顺序输出每个出圈人的编号。

数据范围：
$1\le m,n\le 100$

思路是单链表。由于链表在头部插入是比较方便的，所以我们可以从$n$到$1$顺次插入一个单链表的头部。然后开始模拟题目的过程，每次先走$m-1$步，然后输出下一个位置的节点编号，接着将当前边删掉（如果走$m$步的话，就没法删边了）。只剩一个节点的时候退出循环，把最后剩下的那个节点编号输出即可。代码如下：

#include <iostream>
using namespace std;

const int N = 110;
int n, m;
int h;
int e[N], ne[N], idx;

void init() {
    h = -1, idx = 0;
}

void add_head(int b) {
    e[idx] = b, ne[idx] = h, h = idx++;
}

int main() {
    scanf("%d%d", &n, &m);
    init();
    for (int i = n; i >= 1; i--) add_head(i);
    ne[0] = idx - 1;

    int pos = 0;
    while (ne[pos] != pos) {
        for (int i = 0; i < m - 1; i++) pos = ne[pos];
        printf("%d ", e[ne[pos]]);
        ne[pos] = ne[ne[pos]];
    }

    printf("%d\n", e[pos]);
    return 0;
}

时间复杂度$O(nm)$，空间$O(n)$。