题目描述
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
示例 1:
给定一个链表: 1->2->3->4->5,和 k = 2
返回链表: 4->5
题解
两次遍历法
第一次遍历求单链表表长,第二次遍历直接定位,算法时间复杂度为常数级O(1),空间复杂度为两次单循环O(n)。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* getKthFromEnd(struct ListNode* head, int k){
int length;
struct ListNode* p = head;
for (length = 1; p->next != NULL; p = p->next) {
length ++;
}
for (int i = 1; i <= length-k;i ++) {
head = head->next;
}
return head;
}
快慢指针法
快指针在前一个while循环中从1向后遍历k到2,在第二个while中快指针从2向后遍历length-k到3,同时慢指针从4遍历至5,实现定位,空间复杂度为常数级O(1),时间复杂度为单次遍历O(n)。
1---k---2-----(length-k)-----3
4-----(length-k)-----5---k---6
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* getKthFromEnd(struct ListNode* head, int k){
struct ListNode* slow_p = head, * fast_p = head;
while(k--)
fast_p = fast_p->next;
while(fast_p != NULL) {
slow_p = slow_p->next;
fast_p = fast_p->next;
}
return slow_p;
}