剑指-面试题22 链表中倒数第k个节点

链表中倒数第k个节点

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

思路：

时间复杂度$O(n)$解法：设置两个指针，p1和p2相隔k-1个。p1先走k-1步，之后两个指针一起走，当p1走到链表尾节点时p2即为所求。
特殊情况：
（1）空链表
（2）k=0
（3）链表的长度小于k

C++

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        if(head==nullptr || k==0)
            return nullptr;
        ListNode* p1=head;
        for(int i=0;i<k-1;++i)                          //p1先走到k的位置，再让p2从头开始
        {
            if(p1->next!=nullptr)
                p1 = p1->next;
            else                                        //如果链表长度不大于k，返回空指针
                return nullptr;
        }
        ListNode* p2=head;
        while(p1->next!=nullptr)
        {
            p1 = p1->next;
            p2 = p2->next;
        }
        return p2;

    }
};

python

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        if head==None or k==0:
            return None
        p1=head
        for i in range(1,k):
            if p1.next !=None:
                p1 = p1.next
            else:
                return None
        p2 = head
        while p1.next != None:
            p1 = p1.next
            p2 = p2.next
        return p2