leetcode刷题 (5.7)链表

203

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

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

思路：设置哑结点，这样头结点和其他结点处理一样。

笔记：

C++
链表可以直接赋空值，也可以指向空值。

// 赋空值
ListNode* head = NULL;
// 指向空
p->next = NULL;

Python:
由于Python自动管理内存，所以不需要删除结点直接断线就行，不需要delete操作，也就不需要设tmp。

C++：

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        // 设置虚拟结点
        ListNode* dummy = new ListNode(0);
        // 指向头结点
        dummy->next = head;
        ListNode* p = dummy;
        while(p->next != NULL){
            if(p->next->val == val){
                ListNode* tmp = p->next;
                p->next = p->next->next;
                delete tmp;
            }else{
                p = p->next;
            }
        }
        //不要忘记最后重新赋head，因为原head有可能已经被删了
        head = dummy->next;   
        delete dummy;
        return head;
    }
};

Python：

class Solution:
    def removeElements(self, head: ListNode, val: int) -> ListNode:
        dummy = ListNode(next=head)
        p = dummy
        while(p.next != None):
            if(p.next.val == val):
                p.next = p.next.next
            else:
                p = p.next
        return dummy.next