题目描述
输入一个链表,输出该链表中倒数第k个结点。
解题思路
用两个指针,初始第二个在离第一个的k-1位置处,然后同时移动直到第二个到最后一个结点,此时第一个指针就是指向倒数第k个结点。需要注意传入空指针,没有倒数第k个结点等情况。类似的题有通法,比如第二个指针比第一个指针移动速度快可以得到中间的结点。
Code
/*
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 == nullptr || k == 0) {
return nullptr;
}
ListNode *p = pListHead, *pk = pListHead;
for(int i = 0; i<k-1; i++) {
if(pk->next == nullptr) {
return nullptr;
}
pk = pk->next;
}
while(pk->next != nullptr) {
p = p->next;
pk = pk->next;
}
return p;
}
};
- java
/*
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}*/
public class Solution {
int GetListLength(ListNode head) {
int result = 0;
ListNode pNode = head;
while(pNode != null) {
result++;
pNode = pNode.next;
}
return result;
}
public ListNode FindKthToTail(ListNode head,int k) {
int len = GetListLength(head);
if(len < k) return null;
ListNode p1 = head, p2 = head;
for(int i = 0; i<k; i++) {
p2 = p2.next;
}
while(p2 != null) {
p1 = p1.next;
p2 = p2.next;
}
return p1;
}
}