5.链表（下）——如何轻松写出正确的链表代码

---

前言

写链表代码是最考验逻辑思维能力的。因为，链表代码到处都是指针的操作、边界条件的处理，稍有不慎就容易产生 Bug。

链表代码写得好坏，可以看出一个人写代码是否够细心，考虑问题是否全面，思维是否缜密。所以，这也是很多面试官喜欢让人手写链表代码的原因。

所以，这一节讲到的东西，你一定要自己写代码实现一下，才有效果。

技巧一、理解指针或引用的含义

看懂链表的结构并不是很难，但是一旦把它和指针混在一起，就很容易让人摸不着头脑。所以，要想写对链表代码，首先就要理解好指针。

对于指针的理解，你只需要记住下面这句话就可以了：

将某个变量赋值给指针，实际上就是将这个变量的地址赋值给指针，或者反过来说，指针中存储了这个变量的内存地址，指向了这个变量，通过指针就能找到这个变量。

技巧二：警惕指针丢失和内存泄漏

p->next = x;  // 将p的next指针指向x结点；
x->next = p->next;  // 将x的结点的next指针指向b结点；

初学者经常会在这儿犯错。第一行代码执行完后，第二行其实是自己指向了自己，结点b往后的所有结点都无法访问到了。

以上两点都是菜鸟常见问题，高手可忽略。

技巧三：利用哨兵简化实现难度

C语言在结点p后面插入一个新的结点，只需要两行搞定：

new_node->next = p->next;
p->next = new_node;

但是当向一个空链表中插入第一个结点，上面的代码逻辑就有问题了。应作如下判断：

if (head == null) {
	head = new_node;
}

再来看单链表的删除操作：

p->next = p->next->next;

但是如果要删除链表中的最后一个结点，上面的代码又不能用了。应写成下面的代码：

if (head->next == null) {
	head = null;
}

从前面的一步一步分析，我们可以看出，针对链表的插入、删除操作，需要对插入第一个结点和删除最后一个结点的情况进行特殊处理。这样代码实现起来就会很繁琐，不简洁，而且也容易因为考虑不全而出错。如何来解决这个问题呢？
答案就是引入哨兵结点。

如果我们引入哨兵结点，在任何时候，不管链表是不是空，head 指针都会一直指向这个哨兵结点。我们也把这种有哨兵结点的链表叫带头链表。相反，没有哨兵结点的链表就叫作不带头链表。

哨兵结点是不存储数据。

代码一：

// 在数组a中，查找key，返回key所在的位置
// 其中，n表示数组a的长度
int find(char* a, int n, char key) {
	// 边界条件处理，如果a为空，或者n<=0，说明数组中没有数据，就不用while循环比较了
	if (a == null || n <= 0) {
		return -1;
	}
	
	int i = 0;
	// 这里有两个比较操作：i < n 和 a[i] == key
	while (i < n) {
		if (a[i] == key) {
			return -1
		}
		++i;
	}
	return -1;
}

代码二：

// 在数组a中，查找key，返回key所在的位置
// 其中，n表示数组a的长度
// 我举2个例子，你可以拿例子走一下代码
// a = {4, 2, 3, 5, 9, 6} n=6 key = 7
// a = {4, 2, 3, 5, 9, 6} n=6 key = 6

int find(char* a, int n, char key) {
	if (a == null || n <= 0) {
		return n - 1;
	}
	// 把a[n-1]的值临时保存在变量tmp中，以便之后恢复。tmp=6。 
	// 之所以这样做的目的是：希望find()代码不要改变a数组中的内容
	char temp = a[n - 1];
	// 把key的值放到a[n-1]中，此时a = {4, 2, 3, 5, 9, 7}
	a[n-1] = key;
	
	int i = 0;
	// while 循环比起代码一，少了i
	while (a[i] != key) {
		++i
	}
	
	// 恢复a[n-1]原来的值,此时a= {4, 2, 3, 5, 9, 6}
	a[n-1] = temp;
	
	if (i == n-1) {
		// 如果 i == n-1 说明，在 0...n-2 之间都没有 key，所以返回-1
		return -1;
	} else {
		// 否则，返回i，就是等于key值的元素的下标
		return i;
	}
}

对比两段代码，在字符串 a 很长的时候，比如几万、几十万，你觉得哪段代码运行得更快点呢？答案是代码二，因为两段代码中执行次数最多就是 while 循环那一部分。

第二段代码中，我们通过一个哨兵 a[n-1] = key，成功省掉了一个比较语句 i<n，不要小看这一条语句，当累积执行万次、几十万次时，累积的时间就很明显了。

当然，这只是为了举例说明哨兵的作用，你写代码的时候千万不要写第二段那样的代码，因为可读性太差了。大部分情况下，我们并不需要如此追求极致的性能。

技巧四：重点留意边界条件处理

软件开发中，代码在一些边界或者异常情况下最容易产生bug。要实现没有 Bug 的链表代码，一定要在编写的过程中以及编写完成之后，检查边界条件是否考虑全面，以及代码在边界条件下是否能正确运行。

我经常用来检查链表代码是否正确的边界条件有这样几个：

• 如果链表为空时，代码是否能正常工作？
• 如果链表只包含一个结点时，代码是否能正常工作？
• 如果链表只包含两个结点时，代码是否能正常工作？
• 代码逻辑在处理头结点和尾结点的时候，是否能正常工作？

技巧五：举例画图，辅助思考

你可以找一个具体的例子，把它画在纸上，释放一些脑容量，留更多的给逻辑思考，这样就会感觉到思路清晰很多。比如往单链表中插入一个数据这样一个操作，我一般都是把各种情况都举一个例子，画出插入前和插入后的链表变化，如图所示：

技巧六：多写多练，没有捷径

我精选了 5 个常见的链表操作。你只要把这几个操作都能写熟练，不熟就多写几遍，我保证你之后再也不会害怕写链表代码。

• 单链表反转
• 链表中环的检测
• 两个有序的链表合并
• 删除链表倒数第 n 个结点
• 求链表的中间结点