移除链表元素是我早期刷题接触到的题目,当时对链表这个数据结构不是很熟悉,很多方法都是硬背下来的。但是在回过头来看,现在的理解清晰了许多,将链表的索引与数组的索引进行类比理解,他们有相通之处也有不同的地方。这样对链表的处理就清晰了很多。所以还是要多练才行呀。
移除链表元素,可以用虚节点作为头结点,这样就避免了第一个元素移除时,要重新创建头结点的过程。代码如下所示:
- 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
-
-