zoukankan      html  css  js  c++  java
  • 剑指off:链表中倒数第k个结点

    题目:输入一个链表,输出链表中倒数第k个结点,为了符合大多数人的习惯,本题从1开始计数,即链表的尾结点是倒数第一个结点。例如,一个链表有6个结点,从头结点开始,它们的值依次是1、2、3、4、5、6.这个链表的倒数第三个结点是值为4的结点。

    示例:输入1 -> 2 -> 3 -> 4 -> 5和k=2; 输出: 4 -> 5

    考点:代码的鲁棒性

    算法:1、初始化:前指针former,后指针latter;两个指针均指向头结点

         2、构建双指针距离:前指针向前移动k-1步。

         3、双指针移动:从第k步开始,两个指针一起向前移动,直至former指向尾结点。

       4、返回值:返回latter即可。

    时间复杂度O(N):N为链表长度,former向前走了N-1步,latter向前走了N-k步。

    空间复杂度O(1):双指针former, latter使用常数大小的空间。

    根据以上算法,我们可以写出以下代码:

    # Definition for singly-linked list.
    # class ListNode:
    #     def __init__(self, x):
    #         self.val = x
    #         self.next = None
    
    class Solution:
        def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
            former = latter = head
            for _ in range(k-1):
                former = former.next
            while former.next:
                latter, former = latter.next, former.next
            return latter

    但上述代码的鲁棒性是不够的,存在三个问题:

    1. 链表为空时。由于代码会访问空指针指向的内存,从而造成程序的崩溃
    2. 输入的参数k=0;同样会造成程序的崩溃
    3. 链表的结点个数小于k。同样会存在访问空指针指向的内存而造成程序的崩溃的问题;

    改进代码:

    # Definition for singly-linked list.
    # class ListNode:
    #     def __init__(self, x):
    #         self.val = x
    #         self.next = None
    
    class Solution:
        def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
            # 空链表
            if not head:
                return head
            # k = 0
            if k == 0:
                return None
    
            former = latter = head
            for _ in range(k-1):
                # k > 链表长度
                if former.next is not None:
                    former = former.next
                else:
                    return None
            while former.next:
                latter, former = latter.next, former.next
            return latter
  • 相关阅读:
    图片轮显效果大全
    firefox 对WebRTC支持
    Android AES加密算法及事实上现
    怎样以学习单片机为契机,逐步成为优秀的project师
    HTML中Select的使用具体解释
    POJ 3602 Typographical Ligatures
    远程控制编写之屏幕传输 MFC实现 屏幕截图 发送bmp数据 显示bmp图像
    blend
    POJ3187 Backward Digit Sums
    牛腩公布系统--HTTP 错误 403.14
  • 原文地址:https://www.cnblogs.com/aniu-caili/p/12749957.html
Copyright © 2011-2022 走看看