C++实现学生队列排列算法:插入、删除和输出
C++实现学生队列排列算法:插入、删除和输出
问题描述:
一个学校里老师要将班上n个同学排成一列,同学被编号为 1~n,他采取如下的方法:
- 先将n号同学安排进队列,这时队列中只有他一个人;
- 2~n号同学依次入列,编号为i的同学入列方式为:老师指定编号为 i的同学站在编号为 1~(i-1)中某位同学(即之前已经入列的同学)的左边或右边;
- 从队列中去掉m个同学,其他同学位置顺序不变。
在所有同学按照上述方法队列排列完毕后,老师想知道从左到右所有同学的编号。
输入:
第一行一个整数n,表示了有 n个同学。
第2~n 行,第i行包含两个整数 k,p,其中k为小于i的正整数,p为1或者0。若 p为 0则表示将 i号同学插入到k号同学的左边,p为1则表示插入到右边。
第n+1行为一个整数m ,表示去掉的同学数目。
接下来 m行,每行一个正整数 x表示将x 号同学从队列中移去,如果 x号同学已经不在队列中则忽略这一条指令
输出:
一行,包含最多n个空格隔开的整数,表示了队列从左到右所有同学的编号。
样例输入:
4
1 0
2 1
1 0
2
3
3
样例输出:
2 4 1
C++代码:
#include <iostream>
#include <vector>
using namespace std;
int main() {
int n;
cin >> n;
vector<int> queue;
queue.push_back(n); // 先将n号同学入队列
for (int i = 2; i <= n; i++) {
int k, p;
cin >> k >> p;
if (p == 0) { // 将i号同学插入到k号同学的左边
queue.insert(queue.begin() + k, i);
} else { // 将i号同学插入到k号同学的右边
queue.insert(queue.begin() + k + 1, i);
}
}
int m;
cin >> m;
for (int i = 0; i < m; i++) {
int x;
cin >> x;
// 将x号同学从队列中移除
for (int j = 0; j < queue.size(); j++) {
if (queue[j] == x) {
queue.erase(queue.begin() + j);
break;
}
}
}
for (int i = 0; i < queue.size(); i++) {
cout << queue[i] << " ";
}
return 0;
}
代码解释:
- 使用
vector<int> queue来存储队列中的学生编号。 - 首先将
n号同学加入队列。 - 循环遍历 2~n 号同学,根据输入的
k和p值,使用insert()方法将同学插入到队列的指定位置。 - 循环读取
m个要删除的同学编号,使用erase()方法将该编号的同学从队列中删除。 - 最后循环遍历
queue数组,输出所有学生编号。
优化建议:
- 可以使用更有效的删除算法,例如使用
std::remove()和std::remove_if()结合erase(),以提高删除效率。 - 可以使用更简洁的插入方法,例如使用
queue.insert(queue.begin() + (p == 0 ? k : k + 1), i);来代替if-else语句,使代码更加简洁。 - 可以使用
std::find()方法来查找需要删除的同学编号,避免使用循环遍历。
总结:
本代码使用 C++ 实现了一个模拟学生队列排列的过程,包含插入、删除操作,最终输出队列中所有学生的编号。代码简单易懂,可以作为学习使用 std::vector 的示例。
原文地址: https://www.cveoy.top/t/topic/bzPn 著作权归作者所有。请勿转载和采集!