輸入文件的第一行有兩個整數n,q,分別表示初始鏈表中元素個數和對鏈表操作的次數,第二行有
n個整數,表示初始鏈表的元素,保證這些數字不超過int表示的範圍。
接下來q行,每行可能爲如下兩個操作之一:
1 a b表示在第a個元素後面插入值爲b的元素,
2 a表示將鏈表中第a個元素刪除,如果a大於當時鏈表長度,則忽略此操作
如:
輸入
3 3
1 2 3
1 1 4
1 2 5
2 2
輸出
1 5 2 3
#include <iostream>
#include <list>
using namespace std;
struct Operate {
int a;
int b;
int c;
};
int main()
{
int n, q;
list<int> datalist;
Operate Oper[100];
memset(Oper, 0, sizeof(Oper));
cin >> n >> q;
int data[100];
for (int i = 0; i < n; i++) {
cin >> data[i];
datalist.push_back(data[i]);
}
for (int j = 0; j < q; j++) {
cin >> Oper[j].a;
cin >> Oper[j].b;
if (Oper[j].a == 1) {
cin >> Oper[j].c;
}
}
for (int k = 0; k < q; k++) {
if (Oper[k].a == 1) {
if (Oper[k].b > datalist.size()) {
continue;
}
else {
auto iter = begin(datalist);
advance(iter, Oper[k].b);
datalist.insert(iter, Oper[k].c);
}
}
else if (Oper[k].a == 2) {
auto iter = begin(datalist);
advance(iter, Oper[k].b);
iter--;
datalist.erase(iter);
}
}
for (list<int>::iterator it = datalist.begin(); it != datalist.end(); it++) {
cout << *it << " ";
}
return 0;
}