题目1:在O(1)的时间内删除链表节点。
题目:给定单向链表的头指针和一个节点指针,定义一个函数在O(1)时间内删除该节点。
链表节点定义如下:
class ListNode(object):
def __init__(self, val):
self.val = val
self.next = None
例:
1 -> 2 -> 3 ->4 -> None
head = 1 del = 3
1 -> 2 -> 4
思路
- 关键在于要求O(1)的时间,所以不可以从头遍历。这个题目在leetcode做过,当时还是想了很久。其实并没有删除了节点,而是删除了节点的值,想到这一层,就能明白这个题目的解法。没有删除node的内存,而是将node中保存的数据删除了,因此只需要将node.next.val赋值给node.val,并删除node.next即可。
注意,处理node.next = None的情况,不然会报错None没有val属性。此时必须处理前一节点的next属性,只能从头遍历。
注意,处理删除的节点时头节点,且链表中只有这个节点的情况。Python我不知道如何原地删除头节点…只能做返回None处理,其余情况返回head。- 时间复杂度:平均O(1)
- 空间复杂度:O(1)
代码
class ListNode(object):
def __init__(self, val):
self.val = val
self.next = None
def delete_ListNode(head, delete_node):
"""
:param head: head of NodeList
:param delete_node: node
:return: head
"""
next_node = delete_node.next
if next_node:
# not del end
delete_node.val, next_node.val = next_node.val, delete_node.val
delete_node.next = next_node.next
elif head is not delete_node:
# del end
while head.next is not delete_node:
head = head.next
head.next = None
else:
# delete head
head = None
return head
思考
Leetcode 237. 删除链表中的节点
题目
请编写一个函数,使其可以删除某个链表中给定的(非末尾)节点,你将只被给定要求被删除的节点。
现有一个链表 – head = [4,5,1,9],它可以表示为:
4 -> 5 -> 1 -> 9
示例 1:
输入: head = [4,5,1,9], node = 5
输出: [4,1,9]
解释: 给定你链表中值为 5 的第二个节点,那么在调用了你的函数之后,该链表应变为 4 -> 1 -> 9.
示例 2:
输入: head = [4,5,1,9], node = 1
输出: [4,5,9]
解释: 给定你链表中值为 1 的第三个节点,那么在调用了你的函数之后,该链表应变为 4 -> 5 -> 9.
说明:
链表至少包含两个节点。
链表中所有节点的值都是唯一的。
给定的节点为非末尾节点并且一定是链表中的一个有效节点。
不要从你的函数中返回任何结果。
代码
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution(object):
def deleteNode(self, node):
"""
:type node: ListNode
:rtype: void Do not return anything, modify node in-place instead.
"""
node.val = node.next.val
node.next = node.next.next