【剑指Offer】面试题22. 链表中倒数第k个节点
题目
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
思路
- 先统计链表节点个数
- 计算从前向后移动步数
代码
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
int cnt = 0;
ListNode *cur = head;
while (cur) {
++cnt;
cur = cur->next;
}
int move = cnt - k;
while (move--) {
head = head->next;
}
return head;
}
};