链表中倒数第k个节点
题目
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
思路:
时间复杂度
O
(
n
)
O(n)
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