面试题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步!