这道题没有看解答,自己试着做了一下,处理链表个数小于k的情况时一开始不知道如何处理,就返回头指针;看了不通过的测试用例才明白,直接返回NULL就好了。
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};*/
class Solution {
public:
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
ListNode *pRear = pListHead;
ListNode *pKtoRear = pListHead;
unsigned int i=k; //不需要改成k-1,因为最后pRear会指向NULL,一个空的尾节点
while(i>0 && pRear!=NULL){
pRear = pRear->next;
i--;
}
//链表不足k个节点
if(i!=0) return NULL;
//链表节点数>=k个
while(pRear!=NULL){
pRear = pRear->next;
pKtoRear = pKtoRear->next;
}
return pKtoRear;
}
};
虽然通过了,但是鲁棒性还是不够好,一个比较好的JAVA解答,考虑了输入链表为空、k<=0的情况
/*
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}*/
public class Solution {
public ListNode FindKthToTail(ListNode head,int k) {
if(head==null||k<=0){
return null;
}
ListNode pre=head;
ListNode last=head;
for(int i=1;i<k;i++){ //注意:从1开始
if(pre.next!=null){
pre=pre.next;
}else{
return null;
}
}
while(pre.next!=null){ //注意:判断标准是 pre.next!=NULL,对应到我的C++程序是pre!=NULL
pre = pre.next;
last=last.next;
}
return last;
}
}
还有一个精简的C++写法:
class Solution {
public:
ListNode* FindKthToTail(ListNode* p, unsigned int k) {
//if(!p) return nullptr;
auto p1=p;
for(int i=0;i!=k;++i)
if(!p1)return nullptr;
else
p1=p1->next;
while(p1){
p1=p1->next;
p=p->next;
}
return p;
}
};