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
  • 相关阅读:
    02-17 位图验证码(一般处理程序)+AJAX
    02-18 报表
    SQLite 函数大全
    SQLite中的时间日期函数(转)
    DES,AeS加解密,MD5,SHA加密
    suspendlayout
    AES--高级数据加密标准
    C#中Validating和Validated事件
    Net操作Excel(终极方法NPOI)
    decimal,float和double的区别
  • 原文地址:https://www.cnblogs.com/aniu-caili/p/12749957.html
Copyright © 2011-2022 走看看