链表中倒数第k个节点
题目:输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
解题思路
解法一
- 通过遍历链表统计出节点的个数
- 定义一个指向头节点的指针,移动n-k个结点
- 输出该结点指针
代码如下:
/**
* 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==NULL) return NULL;
if(head->next==NULL) return head;
ListNode* p=head;
ListNode* q=head;;
int n=0,i=0;
//统计出链表的节点数
while(p)
{
p=p->next;
n++;
}
//遍历到n-k节点处
while(q && i<n-k)
{
q=q->next;
i++;
}
return q;
}
};
执行用时:4 ms, 在所有 C++ 提交中击败了90.85%的用户
内存消耗:10.4 MB, 在所有 C++ 提交中击败了42.23%的用户
解法二
- 定义两个节点p、q指针同时指向头节点
- 让p指针先向前移动k个位置,此时它与q指针间隔k个距离
- 在p!=NULL前提下两个指针同时移动
- 当p=NULL时,q刚好指向倒数第k个节点
- 输出q节点指针
代码如下:
/**
* 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==NULL) return NULL;
if(head->next==NULL) return head;
ListNode* p=head;
ListNode* q=head;
int i=0;
//让p先走k步
while(p && i<k)
{
p=p->next;
i++;
}
//p,q同时往后遍历,当p遍历到末尾下一位时,q刚好指向倒数第k个位置
while(p)
{
p=p->next;
q=q->next;
}
return q;
}
};
执行用时:4 ms, 在所有 C++ 提交中击败了90.85%的用户
内存消耗:10.2 MB, 在所有 C++ 提交中击败了93.39%的用户