移除链表元素(leetcode简单篇二百零三题)

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

示例:

输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5


struct ListNode* removeElements(struct ListNode* head, int val)
{
	struct ListNode* cur = head;//当前位置指针
	struct ListNode* prev = NULL;//当前位置之前的指针
	struct ListNode* next;
	while(cur)
	{
		if(cur -> val == val)//判断数值是否需要被删除
		{	
			next = cur -> next;//记住当前位置之后的位置
			free(cur);//释放当前的需要删除的空间
			cur = next;//让cur指向下一个位置
			if(prev == NULL)
			{
				head = next;//此处如果删除的是头节点的话,让头节点指向下一个位置
			}
			else
			{
				prev -> next = next;//如果删除的不是头节点让prev的next的指针指向当前位置
			}
		}
		else//不删除让指针向后走
		{
			prev = cur;
			cur = cur -> next;
		}
	}
	return head;
}

猜你喜欢

转载自blog.csdn.net/lucky52529/article/details/84672692