編程題目：輸入一個鏈表，輸出該鏈表中倒數第k個節點

兩種方法

1.在鏈表的初始化數據中加入 num 數據， 每添加一個節點，num加1，每刪除一個節點，num減1

查找倒數第k個元素，即 指向第一個節點的指針向后移動 num - k 步。

 

2.使用兩個指針 i 和 j， i和j初始化都指向第一個節點。

查看倒數第k個元素，先將 j 向右移動 k-1 步。

再將 i 和 j 同時向右移動，直到 j 指向最后一個元素結束。

這時候 i 指向的元素即 倒數第k個元素。返回 i 指向節點的值即可。

 

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

class Node(object):
        def __init__(self, elem, next_=None):
                self.elem = elem
                self.next = next_

class Simple_List(object):
        def __init__(self):
                self.head = None
                self.num = 0

        def is_empty(self):
                return self.head is None

        def prepend(self, elem):
                self.head = Node(elem, self.head)
                self.num += 1

        def prepop(self):
                if self.is_empty():
                        raise ValueError("List is Empty")
                e = self.head.elem
                self.head = self.head.next
                self.num -= 1
                return e

        def bianli(self):
                p = self.head
                li = []
                while p:
                        li.append(p.elem)
                        p = p.next
                return li

        def find_the_key(self, key):
                n = self.num - key
                p = self.head
                while n:
                        p = p.next
                        n -= 1
                return p.elem

        def find_the_key1(self,key):
                i,j = self.head, self.head
                while key-1:
                        j = j.next
                        key -= 1
                while j.next:
                        i = i.next
                        j = j.next
                return i.elem

if __name__ == "__main__":
        sl = Simple_List()
        for i in [1,4,8,2,4,8,5]:
                sl.prepend(i)
        print("鏈表:",sl.bianli())
        key = int(input("查看該鏈表倒數第k個值:"))
        print("倒數第k個值:",sl.find_the_key(key))
        print("倒數第k個值:",sl.find_the_key1(key))

 

之前寫這個題目的時候沒有注意到邊界問題，當用戶輸入的key值不合理時，會發生錯誤。

所以我們要對用戶輸入的數據進行判斷並處理。

 

修改的地方如下，其他地方不變 

        def find_the_key(self, key):
                if key < 1 or key > self.num:
                        raise ValueError("num must in 0<key<={}".format(self.num))
                n = self.num - key
                p = self.head
                while n:
                        p = p.next
                        n -= 1
                return p.elem

        def find_the_key1(self,key):
                if key < 1:
                        raise ValueError("num is unabled,please repeat input")
                i,j = self.head, self.head
                while key-1:
                        j = j.next
                        key -= 1
                if j is None:
                        raise ValueError("num is unabled,please repeat input")
                while j.next:
                        i = i.next
                        j = j.next
                return i.elem