leetcode 203.移除链表元素

给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足Node.val == val 的节点,并返回 新的头节点 。

示例 1:

leetcode 203.移除链表元素_第1张图片

输入:head = [1,2,6,3,4,5,6], val = 6
输出:[1,2,3,4,5]

示例 2:
输入:head = [], val = 1
输出:[]

示例 3:
输入:head = [7,7,7,7], val = 7
输出:[]

思路:

使用两个指针,prev和cur,cur记录当前指针位置,prev记录前一个位置

如果cur->val!= val,那么prev = cur,cur = cur->next,继续向后查找
leetcode 203.移除链表元素_第2张图片

如果cur->val == val,prev->next = cur->next, 释放cur,将cur置为prev->next
leetcode 203.移除链表元素_第3张图片
leetcode 203.移除链表元素_第4张图片

leetcode 203.移除链表元素_第5张图片

leetcode 203.移除链表元素_第6张图片

还要考虑一种情况,就是头节点的val等于val,我们要删除第一个节点,新的头结点是下一个结点

leetcode 203.移除链表元素_第7张图片
leetcode 203.移除链表元素_第8张图片
leetcode 203.移除链表元素_第9张图片


struct ListNode
{
	int val;
	struct ListNode* next;
};

struct ListNode* removeElements(struct ListNode* head, int val)
{
	struct ListNode* cur = head;
	struct ListNode* prev = NULL;

	while (cur != NULL)
	{
		if (cur->val == val)
		{
			//1.头删
			if (cur == head)
			{
				head = cur ->next;
				free(cur);
				cur = head;
			}
			else
			{
				//2.中间删除
				prev->next = cur->next;

				//释放空间
				free(cur);
				//cur = NULL;
				cur = prev->next;
			}
		}
		else
		{
			//继续向后查找
			prev = cur;
			cur = cur->next;
		}
	}
	
	//返回新的头
	return head;
}

//测试代码
int main()
{
	struct ListNode* n1 = (struct ListNode*)malloc(sizeof(struct ListNode));
	struct ListNode* n2 = (struct ListNode*)malloc(sizeof(struct ListNode));
	struct ListNode* n3 = (struct ListNode*)malloc(sizeof(struct ListNode));
	struct ListNode* n4 = (struct ListNode*)malloc(sizeof(struct ListNode));

	n1->val = 7;
	n2->val = 7;
	n3->val = 7;
	n4->val = 7;

	n1->next = n2;
	n2->next = n3;
	n3->next = n4;
	n4->next = NULL;

	struct ListNode*  head = removeElements(n1,7);

	return 0;
}

你可能感兴趣的