移除链表元素是我早期刷题接触到的题目,当时对链表这个数据结构不是很熟悉,很多方法都是硬背下来的。但是在回过头来看,现在的理解清晰了许多,将链表的索引与数组的索引进行类比理解,他们有相通之处也有不同的地方。这样对链表的处理就清晰了很多。所以还是要多练才行呀。
移除链表元素,可以用虚节点作为头结点,这样就避免了第一个元素移除时,要重新创建头结点的过程。代码如下所示:
class Solution(object):
def removeElements(self, head, val):
dummyhead = ListNode(0)
dummyhead.next = head
cur = dummyhead
while cur.next:
if cur.next.val == val:
cur.next = cur.next.next
else:
cur = cur.next
return dummyhead.next
先讲反转链表
反转链表是很容易出错的一个题,也是我实习面试的时候遇到过的题目,所以印象十分深刻。
首先的要点就是,先链接上前一个元素,再断开与后一个元素的链接。所以需要两个指针,一个指向当前节点,一个指向前一个节点。
想清楚这件事,代码就不难了:
class Solution(object):
def reverseList(self, head):
"""
:type head: ListNode
:rtype: ListNode
"""
# if (head == None or head.next == None):
# return head
# cur = self.reverseList(head.next)
# head.next.next = head
# head.next = None
# return cur
pre, cur = None, head
temp = None
while cur:
temp = cur.next#临时存
cur.next = pre#反转
pre = cur#pre移到反转前的下一站
cur = temp#cur移到反转前的下一站
return pre
注释里面是递归法,思路是一样的
这里面主要是需要熟悉链表的一些方法的使用,代码如下:
class ListNode:
def __init__(self, val=0, next=None):
self.val = val
self.next = next
class MyLinkedList(object):
def __init__(self):
self.dummy_head = ListNode()##添加虚拟头结点,便于删除和添加链表节点
self.size = 0
def get(self, index):
if index < 0 or index >= self.size:##判断下标是否越界
return -1
current = self.dummy_head.next
for i in range(index):
current = current.next
return current.val
def addAtHead(self, val):
self.dummy_head.next = ListNode(val, self.dummy_head.next)
self.size += 1
def addAtTail(self, val):
current = self.dummy_head
while current.next:
current = current.next##通过循环读取到最后个节点
current.next = ListNode(val)
self.size += 1
def addAtIndex(self, index, val):
if index < 0 or index > self.size:
return
current = self.dummy_head
for i in range(index):
current = current.next##通过循环读取到第index+1个节点
current.next = ListNode(val, current.next)
self.size += 1
def deleteAtIndex(self, index):
if index < 0 or index >= self.size:
return
current = self.dummy_head
for i in range(index):
current = current.next
current.next = current.next.next
self.size -= 1