面试题18删除链表的节点
题目
给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点。
返回删除后的链表的头节点。
解题思路
解法1.不设置哨兵节点
通过设置两个节点node and node.next
,在遍历节点的过程中,如果发现node.next.val == val
,则令node.next = node.next.next
.
code1
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def deleteNode(self, head: ListNode, val: int) -> ListNode:
if head.val == val: return head.next # 特殊情况:位于首部,返回head.next
node=head
while node and node.next:#
if node.next.val == val: # 找到了要删除的结点,删除
node.next = node.next.next
node = node.next
return head
解法2:设置哨兵节点
通过设置哨兵节点dummy,使得不用额外判断删除的节点是否为首节点。同时我们将dummy节点的值设为Inf
。
code2:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def deleteNode(self, head: ListNode, val: int) -> ListNode:
dummy = ListNode(float('Inf'))
dummy.next = head
cur = dummy
while cur and cur.next:
if cur.next.val == val:
cur.next = cur.next.next
cur = cur.next
return dummy.next