SWUST OJ 952: 单链表的插入操作的实现 C++实现

题目描述

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

输入

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

输出

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

代码如下：

#include<bits/stdc++.h>
using namespace std;
int m, n, p, a[1005], b[1005], flag = 0;
int main(){
	cin>>m;
	for(int i = 0; i < m; i++) cin>>a[i];
		cin>>n>>p;
	if(n<1||n>m+1){
		 cout<<"error!";
		 return 0;
	}

	for(int i = 0; i <= m; i++){
		if(i<n-1) cout<<a[i]<<" ";
		if(i==n-1) cout<<p<<" ";
		if(i>n-1) cout<<a[i-1]<<" ";
	}
	return 0;	
}