链表中倒数第k个结点

题目描述

输入一个链表，输出该链表中倒数第k个结点。

 

思路

设置两个指针 fast 、slow，fast先走k-1步，然后再一起走；

先走的k-1步中，如果遇到fast=nullptr，说明链表长度小于k-1，就没有倒数第k项，那就直接返回空指针；

当fast走到最后一个结点，slow恰好在倒数第k个结点。

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) :
            val(x), next(NULL) {
    }
};*/
class Solution {
public:
    ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
        if(!pListHead || k<=0) return nullptr;
        ListNode* fast = pListHead;
        ListNode* slow = pListHead;
        for(int i=0;i<k-1;i++){
            fast = fast->next;
            if(!fast) return nullptr;
        }
        while(fast->next){
            fast = fast->next;
            slow = slow->next;
        }
        return slow;
    }
};

pyhton

# -*- coding:utf-8 -*-
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def FindKthToTail(self, head, k):
        # write code here
        res = []
        while head:
            res.append(head)
            head = head.next
        if k<1 or k>len(res):
            return
        return res[-k]