剑指offer

面试题22. 链表中倒数第k个节点

2020-03-13  本文已影响0人  人一己千

题目

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

示例:

给定一个链表: 1->2->3->4->5, 和 k = 2.

返回链表 4->5.

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法

遇事不决双指针,一个指针先走k步,然后一起走。

class Solution(object):
    def getKthFromEnd(self, head, k):
        pHead = head
        for i in range(k):
            if pHead is not None: pHead = pHead.next
            else: return None
        while head and pHead:
            head,pHead = head.next,pHead.next
        return head

这是后来写的,还是觉得上一版代码好看。

class Solution:
    def FindKthToTail(self, head, k):
        # write code here
        headpoint = head
        count = 0
        while headpoint:
            count += 1
            if count > k:
                head = head.next
            headpoint = headpoint.next
        if count < k: return 
        return head  

总结

是走k步!

上一篇下一篇

猜你喜欢

热点阅读