python算法-011找出单链表的倒数第k的元素

孤独的船只,扬起的帆,迷途的人,要去哪儿啊?我敬杯酒,妄想孤单,回家吧~
——林晨阳《船》


题目:给定一个单链表,找出其倒数第k个元素。例如:head->1->3->4->6->7->0->2->4->5->8->3->1,k=3,则倒数第三个元素为8。


今天的题目相比昨天的很简单,但是也有一定的技巧。先来分析下题目:给定的是单链表,只能从头向后遍历,因此他要查倒数第k个,也还是要从头开始查。要求是倒数第k个,那么如何判断当前的节点是否是倒数第k个,就是我们必须要解决的问题。一般我们先想到的是:先遍历一次链表,算出链表的长度N,那么我们求倒数第k个的问题,就可以转换为——求正序第N-k+1个元素。这个算法实现很简单,但是时间复杂度为O(N^2),这不是我们希望看到的。后面我会介绍一种快的方法,先来看一下这个——"转换问题法"如何用代码实现:

#函数开始
#输入链表头以及k值
def FindLastButK_1(head,k):
    #如果链表非空则执行
    if head.next is not None:
        cur=head #因为这里是计算链表长度,需要将第一个节点也算上
        i=0 #用于记录链表长度
        while cur.next is not None:
            i+=1
            cur=cur.next
        #将问题转化为求 正序数第n-k+1个元素
        n=i-k+1
        #遍历到正序第N-k+1个元素
        cur=head
        while n>0:
            cur=cur.next
            n=n-1
        #返回要求的元素
        return cur.data
    return None

这算代码非常简单,没有什么难度,加上这么详细的注释,我的室友都能看懂了。
下面来看一下一个比较新奇的方法,在学习之前我反正没想到:
用两个指针——slow和fast,从名字上看就是——快慢指针法。那么如何一个快慢指针呢?接着看->
在开始遍历时fast比slow快k-1步,比如k=3,那么
slow=head.next
fast=slow.next.next
然后slow与fast一起向前遍历,每次一个节点,直到fast到达最后一个节点。此时的slow刚好到达倒数第K个元素。
看下面的字符图:

K=4
  1->3->4->6->7->0->2->4
1-^  ^  ^  ^  ^  ^  ^  ^
2-|  |  |  |  |  |  |  |
3-s  |  |  f  |  |  |  |
4-   s  |  |  f  |  |  |
5-      s  |  |  f  |  |
6-         s  |     f  |
7-            s        f

这样是不是很好理解。只需要在开头处理下slow和fast,就可以实现找到倒数第k个元素。下面我们来用代码实现这里本来想给两个版本,一个用循环处理,一个用exec语句,但是exec我不会用,一用就错:

#用循环处理的
def FindLastButK_2(head,k):
    i=0
    slow=head.next
    fast=head.next
    #这里用一个循环来处理fast
    while k-1>0:
        fast=fast.next
        k-=1
    #我本来想用exec语句完成的但是我不会,怎么写都不对,很难受
    #以后有空我会加上的

    #判断链表是否为空,不空开始遍历,每次一步
    if head.next is not None:
        while fast.next is not None:
            slow=slow.next
            fast=fast.next
        #当fast到达最后一个节点时,slow刚好是倒数第K个
        return slow.data
    return None

下面来看下程序运行:

if __name__ == '__main__':
    head=creatLink(10)
    print("head:")
    cur = head.next
    while cur != None:
        print(cur.data)
        cur = cur.next
    k=int(input("请输入k:\n"))
    item1=FindLastButK_1(head,k)
    item2=FindLastButK_2(head,k)
    print("FindLastButK_1\n链表倒数第k个元素为:",item1)
    print("FindLastButK_2\n链表倒数第k个元素为:",item2)

运行结果:


3

5

10

我把两种方法都运行了。


全部代码:

import random
class LNode:
    def __init__(self,arg):
        self.data=arg
        self.next=None

"""
题目描述:
给定链表Head->1->1->3->3->5->7->7->8
单链表倒数第k=3个元素为7
要求:
方法:遍历链表计算链表长度n,然后求顺序第n-k+1个元素即可,再遍历一遍链表就可以得到结果
"""
#先构造链表
def creatLink(x):
    i = 1
    head = LNode(None)
    tmp = None
    cur = head
    while i <= x:
        n = random.randint(1, 9)
        tmp = LNode(n)
        cur.next = tmp
        cur = tmp
        i += 1
    return head
#函数开始
#输入链表头以及k值
def FindLastButK_1(head,k):
    #如果链表非空则执行
    if head.next is not None:
        cur=head #因为这里是计算链表长度,需要将第一个节点也算上
        i=0 #用于记录链表长度
        while cur.next is not None:
            i+=1
            cur=cur.next
        #将问题转化为求 正序数第n-k+1个元素
        n=i-k+1
        cur=head
        while n>0:
            cur=cur.next
            n=n-1
        return cur.data
    return None

def FindLastButK_2(head,k):
    i=0
    slow=head.next
    fast=head.next
    while k-1>0:
        fast=fast.next
        k-=1
    #exec('''"fast=head.next"+".next"*(k-1)''')
    if head.next is not None:
        while fast.next is not None:
            #print("s",slow.data)
            #print("f",fast.data)
            slow=slow.next
            fast=fast.next
        return slow.data
    return None
if __name__ == '__main__':
    head=creatLink(10)
    print("head:")
    cur = head.next
    while cur != None:
        print(cur.data)
        cur = cur.next
    k=int(input("请输入k:\n"))
    item1=FindLastButK_1(head,k)
    item2=FindLastButK_2(head,k)
    print("FindLastButK_1\n链表倒数第k个元素为:",item1)
    print("FindLastButK_2\n链表倒数第k个元素为:",item2)

今天的算法就到这里,明天写爬虫。
下午学习的时候,遇到了一个。。。不算bug的bug,是我自己学的不扎实。我想在函数中更改全局变量的值,但是并没有告诉函数,那是个全局变量。电脑就把他当成局部变量了,不巧的是我还递归的调用了该函数,于是就出现了各种问题,我想了各种办法——把函数写成类的一个方法,把全局变量作为类的固有属性、把全局变量便成局部变量·····总之我是越来越烦,然后越来越错,错的更离谱!!!
我睡了一觉,醒过来我在最初的函数第一行加了一句global x.........解决了。

大家有什么需要都可以来找我,简书号、微信公众号:Dkider 私信、简信,都可以找到我。
更多的问题以及文章源码见github

孤独的人啊,不要悲伤,也许明天就看见了太阳。时光漫长,珍惜很短,别想了,跟我回家吧~
——林晨阳《船》

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,378评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,356评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,702评论 0 342
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,259评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,263评论 5 371
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,036评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,349评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,979评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,469评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,938评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,059评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,703评论 4 323
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,257评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,262评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,485评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,501评论 2 354
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,792评论 2 345