删除链表中的倒数第n个节点—Python
单链表的倒数第K个节点
- 首先从前往后寻找单列表的第K个节点很简单,只需要一个for循环
- 而如何查找从后往前的第K个节点?一般算法题只是会给一个ListNode头结点表示一条单链表,所以不能得出这条链表的长度n,而是需要先遍历一遍链表算出n的值,在遍历链表计算n-k的值。— 这种需要两次遍历(效率太差)
- 只遍历一次链表:就算出倒数第K个结点,如下:
假设K = 2, 思路如下:
首先,我们先让一个指针p1指向链表的头节点head,然后走k步:
现在的p1,只要再走n - k步,就能走到链表末尾的空指针了对吧?再用一个指针p2指向链表头节点head:
接下来让p1和p2同时向前走,p1走到链表末尾的空指针时走了n - k步,p2也走了n - k步,也就恰好到达了链表的倒数第k个节点:
只遍历了一次链表,就获得了倒数第k个节点p2。
代码逻辑如下:
// 返回链表的倒数第 k 个节点
ListNode findFromEnd(ListNode head, int k) {
ListNode p1 = head;
// p1 先走 k 步
for (int i = 0; i < k; i++) {
p1 = p1.next;
}
ListNode p2 = head;
// p1 和 p2 同时走 n - k 步
while (p1 != null) {
p2 = p2.next;
p1 = p1.next;
}
// p2 现在指向第 n - k 个节点
return p2;
}
有了上面的思想,再来看leedcode上面这道题目,删除链表中的倒数第n个结点
直接进行代码求解就是:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution(object):
def removeNthFromEnd(self, head, n):
# 创建一个虚节点,然后
dummy = ListNode(0, head)
dummy.next = head
# 调用函数返回倒数第n + 1个值即可
node = self.findLNode(dummy, n + 1)
# 进行链表节点的删除操作
node.next = node.next.next
return dummy.next
# 找到倒数第n个节点
def findLNode(self, head, n):
# 设置一个节点指向头结点,直接进行指向即可
first = head
i = 0
# 先走n步
while i < n:
first = first.next
i += 1
# 设置第二个节点进行跟踪操作
second = head
# 在同时走总长度- k步数,
while first is not None:
first = first.next
second = second.next
# 直接进行返回即可
return second