链表第二天：虚拟头节点

leetcode第203题：移除链表元素

题意：删除链表中等于给定值 val 的所有节点。

思路

这里以链表 1 4 2 4 来举例，移除元素4。

如果使用C，C++编程语言的话，不要忘了还要从内存中删除这两个移除的节点， 清理节点内存之后如图：

无论使用哪种语言，都应该注意对内存的管理。

这种情况下的移除操作，就是让节点next指针直接指向下下一个节点就可以了，

那么因为单链表的特殊性，只能指向下一个节点，刚刚删除的是链表的中第二个，和第四个节点，那么如果删除的是头结点又该怎么办呢？

这里就涉及如下链表操作的两种方式：

①直接使用原来的链表来进行删除操作。
②设置一个虚拟头结点在进行删除操作。

来看第一种操作：直接使用原来的链表来进行移除。

移除头节点和移除其他节点的操作是不一样的，因为链表的其他节点都是通过前一个节点来移除当前节点，而头结点没有前一个节点。

所以头结点如何移除呢，其实只要将头结点向后移动一位就可以，这样就从链表中移除了一个头结点。

依然别忘将原头结点从内存中删掉。

这样移除了一个头结点，是不是发现，在单链表中移除头结点 和 移除其他节点的操作方式是不一样，其实在写代码的时候也会发现，需要单独写一段逻辑来处理移除头结点的情况。

那么可不可以 以一种统一的逻辑来移除 链表的节点呢。

其实可以设置一个虚拟头结点，这样原链表的所有节点就都可以按照统一的方式进行移除了。

来看看如何设置一个虚拟头。依然还是在这个链表中，移除元素1。

这里来给链表添加一个虚拟头结点为新的头结点，此时要移除这个旧头结点元素1。

这样是不是就可以使用和移除链表其他节点的方式统一了呢？

来看一下，如何移除元素1 呢，还是熟悉的方式，然后从内存中删除元素1。

最后呢在题目中，return 头结点的时候，别忘了 return dummyNode->next;， 这才是新的头结点

代码实现（c++）

在原链表上操作

class solution
{
public:
	//ListNode 昨天认识链表中的链表节点的定义
	ListNode* removeElements(ListNode* head, int val)
	{
		//头节点单独处理
		while (head != NULL && head->val == val)
		{//注意为什么使用while 而不是 if。因为要删除链表中所有值 == val 的节点，所有不排除前两个甚至多个值都为val的，像[1, 1, 1, 4, 5]，要移除值为1的节点。
			//使用临时指针指向头节点
			TreeNode* tmp = head;
			//头节点后移
			head = head->next;
			//释放要删除的头节点的内存
			delete tmp;
		}
		//移除非头节点
		TreeNode* cur = head;
		while (cur != NULL && cur->next != NULL)
		{
			if (cur->next->val == val)
			{
				//找到要删除节点，使用临时指针指向释放内存
				TreeNode* tmp = cur->next;
				cur->next = cur->next->next;
				delete tmp;
			}
			else
			{
				//未找到，后移
				cur = cur->next;
			}
		}
		return head;
	}
};

使用虚拟头节点移除元素

class solution
{
public:
	TreeNode* removeElements(TreeNode* head, int val)
	{
		//定义一个虚拟头节点
		TreeNode p;	//这里调用编译器自动生成的默认构造
		p.next = head;	//让这个虚拟头节点指向头节点
		TreeNode* cur = &p;
		while (cur->next != NULL)
		{
			if (cur->next->val == val)
			{
				TreeNode* tmp = cur->next;
				cur->next = cur->next->next;
				delete tmp;	
			}	
			else
			{
				cur = cur->next;	
			}
		}
		return p.next;	//返回虚拟头节点的next指向的头节点。
		//注意前面定义的虚拟头节点是一个局部节点变量，程序结束后自动释放内存，并没有造成内存泄漏。
	}
};

总结

使用虚拟头节点操作链表确实是方便许多。
还需要注意的是，操作链表时都是用着临时指针变量的，为什么不直接用头节点指针操作？前面提过：单链表的特殊性，只能指向下一个节点，一旦改变头节点指针指向，前面的节点就找不到了，即没有指向前面节点的指针了。没有指针指向的内存，就是内存泄漏了。

所以后面操作链表时都应使用临时指针变量，这算是对做过的链表题目的一点理解吧。

转自：链表：听说用虚拟头节点会方便许多？