PTA——链表——答案与分析

7-4 约瑟夫环

N个人围成一圈顺序编号，从1号开始按1、2、3......顺序报数，报p者退出圈外，其余的人再从1、2、3开始报数，报p的人再退出圈外，以此类推。
请按退出顺序输出每个退出人的原序号。

输入格式:

输入只有一行，包括一个整数N(1<=N<=3000)及一个整数p(1<=p<=5000)。

输出格式:

按退出顺序输出每个退出人的原序号，数据间以一个空格分隔，但行尾无空格。

输入样例:

在这里给出一组输入。例如：

7 3

输出样例:

3 6 2 7 5 1 4

 

#include<iostream>
#include<vector>
using namespace std;
int main() {
    long long N, p;
    cin >> N >> p;
    // 使用循环链表来表示圈子
    vector<int> circle(N);
    for (int i = 0; i < N; i++) {
        circle[i] = i + 1;
    }

    int current = 0; // 当前报数的人在圈子中的索引
    for (int i = 0; i < N - 1; i++) {//这个地方之所以是n-1，是只循环n-1次，直到只剩下一个人为止，最后输出的a[0]就是最后剩下的那一个人
        // 计算下一个要退出的人的位置
        current = (current + p - 1) % (N - i);
        // 输出当前退出人的编号
        cout << circle[current] << " ";
        // 从圈子中删除当前退出的人
        circle.erase(circle.begin() + current);
    }

    // 输出最后一个退出的人的编号
    cout << circle[0] << endl;

    return 0;
}