题目:
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。
例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
法1:最直接的寻找方式,遍历两遍链表。第一遍记录链表的长度,第二遍找到倒数第k个节点
倒数第k个节点为:n = n - k + 1;
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
if(head == NULL) return head;
int n = 0;
ListNode *p = head;
while(p != NULL)
{
n++;
p = p->next;
}
n = n - k + 1;
cout << n;
if(n < 0) return NULL;
p = head;
while(n - 1)
{
p = p->next;
n--;
}
return p;
}
};
法2:只遍历一遍,定义一个前节点front,和后节点back
前节点先走k步,然后后节点和前节点一起往前走,当前节点为NULL时,即后节点为倒数第k个节点。
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
if(head == NULL) return head;
ListNode *front = head;
ListNode *behind = head;
for(int i = 0; i < k; i++)
{
front = front->next;
if(front == NULL) return head;
}
while(front != NULL)
{
front = front->next;
behind = behind->next;
}
return behind;
}
};