SWUST OJ#952 单链表插入

目录

题目

思路

代码

数据结构

STL

---

题目

题目描述

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

输入

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

输出

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

样例输入

5
1 2 3 4 5
3
6

样例输出

1 2 6 3 4 5

思路

比较简单，就是从头遍历链表，找到输入位置的 地址p，在p位置进行头插就行。

// 单链表插入
void LinklistInsert(Linklist* &L) {
    Linklist *p=L;
    int pos;cin>>pos;//输入位置
    //输入不合法退出程序
    if(pos>LinklistSize(L)||pos<=0) {
        cout<<"error!";
        exit(0);//退出程序
    }
    //找到插入的地址 p
    for(int i=1;i<pos;i++) p=p->next;
    //创建新节点q
    Linklist *q;
    LinklistInit(q);
    cin>>q->data; //输入新节点q
    q->next=p->next;//链接节点q
    p->next=q;//覆盖位置p
}

代码

数据结构

链表模板

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
//定义链表
typedef struct LNode {
    int data;
    LNode* next;
}Linklist;
// 初始化链表
void LinklistInit(Linklist* &L) {
    L=(Linklist*)malloc(sizeof(Linklist));
    L->next=NULL;
}
//输入数据，创建链表
void LinklistCreate(Linklist* &L) {
    int n;cin>>n;
    Linklist *p=L;
    while(n--) {
        Linklist* q;
        LinklistInit(q);
        cin>>q->data;
        p->next=q;
        p=q;
    }
}
int LinklistSize(Linklist* &L) {
    int size=0;
    Linklist *p=L;
    while(p->next!=NULL) {
        size++;
        p=p->next;
    }
    return size;
}
// 单链表插入
void LinklistInsert(Linklist* &L) {
    Linklist *p=L;
    int pos;cin>>pos;
    if(pos>LinklistSize(L)||pos<=0) {
        cout<<"error!";
        exit(0);
    }
    while(--pos) {
        p=p->next;
    }
    Linklist *q;
    LinklistInit(q);
    cin>>q->data;
    q->next=p->next;
    p->next=q;
}
//输出数据，打印链表
void LinklistPrint(Linklist* &L) {
    L=L->next;
    while(L!=NULL) {
        cout<<L->data<<" ";
        L=L->next;
    }
}
int main() {
    Linklist *L;
    LinklistInit(L);
    LinklistCreate(L);
    LinklistInsert(L);
    LinklistPrint(L);
    return 0;
}

STL

#include <bits/stdc++.h>
using namespace std;
list<int>a;
int main(){
	int n,x;cin>>n;
	while(cin>>x,a.push_back(x),--n);
	int pos,data;cin>>pos>>data;
	if(pos>a.size()||pos<=0) cout<<"error!";
	else{
		list<int>::iterator it=a.begin();
		for(int i=1;i<pos;i++) it++;
	    a.insert(it,data);
	    for(it=a.begin();it!=a.end();it++) cout<<*it<<" ";	
	}
    return 0;
}