SWUSTOJ #952 单链表的插入操作的实现

题目

建立长度为 n 的单链表，在第 i 个结点之前插入数据元素 data。

输入

第一行为自然数n，表示链式线性表的长度；
第二行为n个自然数表示链式线性表各元素值；
第三行为指定插入的位置i；第四行为待插入数据元素data。

输出

指定插入位置合法时候，输出插入元素后的链式线性表的所有元素，元素之间用一个空格隔开。输入不合法，输出"error!"。

样例输入

5
1 2 3 4 5
3
6

样例输出

1 2 6 3 4 5

源代码

#include 
#include 
#include 

using namespace std;

int main()
{
     
    list<int> L;
    int n;
    cin >> n;
    int arr[1000];
    for(int i=0; i<n; i++)
    {
     
        cin >> arr[i];
        L.push_back(arr[i]);
    }
    int m;
    cin >> m;
    int date;
    cin >> date;
    if(m > L.size())
    {
     
        cout << "error!";
        return 0;
    }
    list<int>::iterator iter;
    iter = L.begin();
    for(int i=1; i<m; i++)
    {
     
        iter++;
    }
    L.insert(iter, date);
    iter = L.begin();
    for(iter = L.begin(); iter != L.end(); iter++)
    {
     
        cout << *iter << " ";
    }
    return 0;
}