leetcode 203.移除链表元素
题目描述
删除链表中等于给定值 val 的所有节点。
示例:
输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5
解题思路
1.删除链表中节点的方法
链表中可能包含多个val值,首先如果链表头等于val值得时候,直接是头结点向后移动一位,删除当前的节点;如果中间或者尾结点中的值等于val,使用下一个节点把当前节点进行替换就可以删除当前节点。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
// 链表是否为空
if(head == NULL){
return NULL;
}
// 删除头结点为val的情况
while(head && head->val == val){
head = head->next;
}
// 头结点不为val的情况
ListNode* cur = head; // 定义cur移动节点
while(cur){
if(cur->next && cur->next->val == val){ // 下一个节点不为空,且值等于val,那么删除该节点
cur->next = cur->next->next;
}
else{
cur = cur->next; // 移动节点
}
}
return head;
}
};
2.建立新链表的方法
创建一个新的链表newHead,如果head中的节点值不等于val,就把当前节点添加到newHead的尾部。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
ListNode* newHead = new ListNode(-1); // 定义一个新的链表
ListNode* pre = newHead; // pre变量用于移动新链表中的节点
ListNode* cur = head; // 移动head中的节点
while(cur){
if(cur->val != val){ // 如果当前节点的值不等于val,把当前节点添加到newHead中
pre->next = cur;
pre = pre->next; // 移动pre到newHead的最后一个节点,以添加下一个满足条件的节点
cur = cur->next; // 移动head节点
}
else{
cur = cur->next; // 移动head节点
}
}
pre->next = NULL; // 尾部处理,当最后一个节点的值为val时,while中未做处理,pre下一个节点的值不是NULL而是含有val值得节点,所以需要使pre的下一个节点的值为NULL;如果尾结点的值不为NULL,那么pre下一个节点的值就是NULL,再次赋值并不影响
return newHead->next;
}
};
欢迎大家关注我的个人公众号,同样的也是和该博客账号一样,专注分享技术问题,我们一起学习进步