【CSP试题回顾】201703-2-学生排队（优化）

CSP-201703-2-学生排队

解题思路

这段代码的目的是模拟学生在队列中进行多次位置调整的过程，以达到体育老师小明所希望的最终队列排序。代码逻辑分为以下几个步骤：

1. 初始化队列：

   • 代码首先通过读取输入n（学生数量），来初始化一个名为student的vector<int>，该向量存储了按学号从小到大排列的学生。这是通过一个for循环实现的，循环变量i从0开始到n-1，对应的student向量的元素被初始化为i+1，这样就得到了一个按学号顺序排列的队列。

2. 读取调整指令：

   • 然后，代码读取另一个输入m，表示需要进行的调整次数。接下来的m行中，每行包含两个整数p和q。这里，p表示需要调整位置的学生的学号，而q表示移动的距离（如果q为正，向后移动；如果q为负，向前移动）。

3. 执行调整操作：

   • 对于每个调整指令，代码首先找到需要调整的学生当前在队列中的位置（索引），这是通过一个for循环完成的，遍历student向量直到找到学号为p的学生。
   • 根据q的值（正或负），决定是向前还是向后移动。然后，计算出插入位置的索引，并使用vector的insert方法在正确的位置插入学号p，随后，原来位置上的学号p被移除，保证了学号p的学生按照要求移动了正确的距离。

4. 输出最终队列：

   • 最后，使用一个for循环遍历调整后的student向量，并输出每个学生的学号，得到最终的队列。

完整代码

#include <iostream>
#include <algorithm>
#include <iomanip>
#include <vector>
#include <string>
using namespace std;

int n, m, p, q;

int main()
{
    cin >> n >> m;
    vector<int> student(n);
    for (size_t i = 0; i < n; i++)
    {
        student[i] = i + 1;
    }
    for (size_t i = 0; i < m; i++)
    {
        cin >> p >> q;
        int index = 0;
        for (size_t i = 0; i < n; i++) // find index
        {
            if (student[i] == p) 
            {
                index = i;
                break;
            }
        }
        // insert new student
        if (q > 0) // move back
        {
            vector<int>::iterator it = student.begin() + index + q + 1;
            student.insert(it, p);
            student.erase(student.begin() + index);
        }
        else // move forward
        {
            vector<int>::iterator it = student.begin() + index + q;
            student.insert(it, p);
            student.erase(student.begin() + index+1);
        }
    }
    for (auto &it : student)
    {
        cout << it << " ";
    }
    return 0;
}