(C语言 C++) 数据结构：单链表的实现

#include <iostream>
#include <algorithm>

using namespace std;

#define ERROR 0
#define OK 1

typedef bool Status;
typedef int EleType;

typedef struct Lnode
{
	EleType val;
	Lnode* next;
} *LinkList; //LinkList类型指向Lnode

//打印链表
void printList(LinkList& L)
{
	LinkList p = L;
	while (p->next)
	{
		p = p->next;
		cout << p->val << " ";
	}
	cout << endl;
}

//初始化
Status init(LinkList& L)
{
	L = new Lnode; //头指针指向头结点
	L->next = nullptr; //头结点next域赋为空
	return OK;
}
//前插法创建单链表
void createList_H(LinkList& L, int n)
{
	for (int i = 0; i < n; i ++ )
	{
		LinkList p = new Lnode;
		cin >> p->val;
		p->next = L->next;
		L->next = p;
	}
}
//后插法创建单链表
void createList_R(LinkList& L, int n)
{
	LinkList tail = L; //tail是链表尾指针

	for (int i = 0; i < n; i ++ )
	{
		LinkList p = new Lnode;
		cin >> p->val;
		p->next = nullptr;
		tail->next = p;
		tail = tail->next;
	}
}
//查找元素
LinkList locate(LinkList L, EleType e)
{
	LinkList p = L->next; //指针指向L的首元节点
	while (p && p->val != e) p = p->next;

	return p; //查找到元素，返回非空指针。没有查找到元素，返回空指针
}
//在第i个位置插入元素
Status insert(LinkList& L, int i, EleType e)
{
	LinkList p = L; //p指向头结点
	int j = 0;

	//找到第i-1个结点
	//p用来检测 i > n + 1, j < i - 1用来检测 i < 1
	while (p && j < i - 1)
	{
		p = p->next;
		j ++ ;
	}

	if (!p || i < 1) return ERROR; //p为空指针，i > n + 1

	LinkList tmp = new Lnode();
	tmp->val = e;
	tmp->next = p->next;
	p->next = tmp;

	return OK;
}
//在第i个位置删除元素
Status del(LinkList& L, int i)
{
	LinkList p = L;
	int j = 0;
	
	//找到第i-1个结点
	//p用来检测 i > n, j < i - 1用来检测 i < 1
	while (p->next && j < i - 1)
	{
		p = p->next;
		j ++ ;
	}

	if (!p->next || i < 1) return ERROR; //p->next为空，i > n

	LinkList tmp = p->next; //保存第i个节点
	p->next = tmp->next;
	delete tmp;

	return OK;
}
//两个有序单链表的归并运算
void merge(LinkList& L1, LinkList& L2, LinkList& Res)
{
	//创建指针p1,p2分别指向L1,L2的首元节点。创建指针p3指向Res的头节点, 保证p3一直是Res的尾指针
	LinkList p1 = L1->next, p2 = L2->next, p3 = Res;

	//有一条链表遍历到末尾就停止循环
	while (p1 && p2)
	{
		if (p1->val <= p2->val) 
		{
			p3->next = p1; //将p1指向的元素插入Res
			p1 = p1->next; //p1指针向后移
		} 
		else
		{
			p3->next = p2;
			p2 = p2->next;
		}
		p3 = p3->next; 
	}

	//将没扫完的那条链表的剩余部分接到Res的最后
	p3->next = p1 ? p1 : p2;
}
//单链表的建立，查找，插入，删除函数测试
void test1();
//两个有序单链表的归并运算测试
void test2();

int main()
{
	cout << "单链表的建立，查找，插入，删除函数的测试：" << endl;
	test1();
	cout << "两个有序单链表的归并运算测试：" << endl;
	test2();

	return 0;
}
//单链表的建立，查找，插入，删除函数测试
void test1()
{
	LinkList L;
	init(L);

	//后插法给链表添加十个数
	cout << endl;
	cout << "建立单链表函数测试：" << endl;
	cout << "后插法给链表添加十个数：";
	createList_R(L, 10);

	cout << "当前链表：";
	printList(L);
	cout << endl;

	//插入，删除，查找函数测试
	cout << "插入函数测试：" << endl;
	cout << "在第二个位置插入10：";
	insert(L, 2, 10);
	printList(L);
	cout << "在最后一个结点后面尾插100：";
	insert(L, 12, 100);
	printList(L);
	cout << "在不合法的位置(-1)插入200：";
	if (!insert(L, -1, 200)) cout << "该位置不合法！" << endl;
	cout << "在不合法的位置(20)插入100：";
	if (!insert(L, 20, 100)) cout << "该位置不合法！" << endl;
	cout << endl;

	cout << "删除函数测试：" << endl;
	cout << "删除第一个结点：";
	del(L, 1);
	printList(L);
	cout << "删除最后一个结点：";
	del(L, 11);
	printList(L);
	cout << "删除不合法的位置(-1)：";
	if (!del(L, -1)) cout << "该位置不合法！" << endl;
	cout << "删除最后一个结点的后一个结点(11)：";
	if (!del(L, 11)) cout << "该位置不合法！" << endl;
	cout << endl;

	cout << "查找函数测试：" << endl;
	cout << "查找元素10：";
	if (locate(L, 10)) cout << "10存在" << endl;
	cout << "查找元素1000：";
	if (!locate(L, 1000)) cout << "1000不存在！" << endl;
	cout << endl;
}
//两个有序单链表的归并运算测试
void test2()
{
	LinkList L1, L2, Res;
	init(L1), init(L2), init(Res);
	LinkList p1 = L1, p2 = L2; //L1，L2的尾指针

	srand(time(0));

	int x1[10], x2[10];
	for (int i = 0; i < 10; i ++ ) x1[i] = rand() % 10 + 1;
	for (int i = 0; i < 10; i ++ ) x2[i] = rand() % 10 + 1;

	sort(x1, x1 + 10), sort(x2, x2 + 10);

	for (int i = 0; i < 10; i++)
	{
		LinkList p = new Lnode();
		p->val = x1[i], p->next = nullptr;
		p1->next = p;
		p1 = p1->next;
	}

	for (int i = 0; i < 10; i++)
	{
		LinkList p = new Lnode();
		p->val = x2[i], p->next = nullptr;
		p2->next = p;
		p2 = p2->next;
	}

	cout << "初始L1：";
	printList(L1);
	cout << "初始L2：";
	printList(L2);
	cout << "归并后：";
	merge(L1, L2, Res);
	printList(Res);
}