题目:
输入一个链表,输出该链表中倒数第k个结点。
这道题虽然还算简单但我还是没思路:)
最简单的想法就是,先遍历一遍整个链表,计算出链表的长度,然后再遍历到n-k的位置,取出倒数第k个节点。
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};*/
class Solution {
public:
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
if (!pListHead || k <= 0) {
return NULL;
}
ListNode* p1 = pListHead;
int len = 0;
while (p1) {
len++;
p1 = p1->next;
}
if (k > len) {
return NULL;
}
ListNode* p2 = pListHead;
for (int i = 0; i < len - k; i++) {
p2 = p2->next;
}
return p2;
}
};
最优解可以不用提前遍历完链表就得出。这里还是用到了一点点快慢指针的思想吧,先设置一个快指针,从头节点开始走k-1步,然后设置一个慢指针从头节点开始,两个指针同步向前,直到快指针指向最后一个节点时,慢指针所指向的节点就是所求的倒数第k个节点。这个做法的精髓就在于,快指针和慢指针之间的差距一直控制在k-1。代码如下:
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};*/
class Solution {
public:
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
if (!pListHead || k <= 0) {
return NULL;
}
ListNode* p1 = pListHead;
for (int i = 0; i < k - 1; i++) {
if (p1->next) {
p1 = p1->next;
}
else {
return NULL;
}
}
ListNode* p2 = pListHead;
while (p1->next) {
p1 = p1->next;
p2 = p2->next;
}
return p2;
}
};
这道题其实思路挺简单的,但是要注意考虑全面,思考几个特殊情况:
1. 空链表
2. k比链表的长度还长
3. k=0
另外把这道题扩展一下,如果要求出链表的中间节点,即当链表长度为奇数时返回正中间的节点,当链表长度为偶数时返回中间两个节点之中的任意一个,也可以采用快慢指针的思想:快指针每次走两步,慢指针每次走一步。