题目描述:输入一个链表,输出该链表中倒数第k个结点。
时间限制:1秒 空间限制:32768K
Ⅰ 依然是我那通过了的笨方法
思路:两次遍历,记录节点位置再返回
代码:
/*
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==NULL)
return pListHead;
ListNode* head = pListHead;
ListNode* dummy = new ListNode(0);
ListNode* pre = dummy;
pre->next = pListHead;
int num = 0,count = 0;
while(head!=NULL){
num++;
head = head->next;
}
int pRes = num-k;
if(pRes==0)
return pListHead;
while(pListHead!=NULL){
pListHead = pListHead->next;
count++;
if(count==pRes)
break;
}
return pListHead;
}
};
Ⅱ 一次遍历的方法
这个答案的思路其实很聪明,但是自己写了一遍出现了错误,后来发现错误的原因是循环里用了k–和判断k是否小于1,而题目定义的k是无符号int,因此出现了错误。
思路:类似快慢指针遍历链表,定义两个指针,当第一个指针跑了k-1个时,第二个指针从头开始跑,这样第一个指针跑到末尾时第二个指针指向的就是倒数第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==NULL)
return pListHead;
//两个指针都指向头结点
ListNode* p = pListHead;
ListNode* pre = pListHead;
//记录k值,注意这里将无符号转成有符号
int a = k;
//记录节点个数
int count = 0;
//p指针先跑,并且记录节点数,当p指针跑了k-1个节点后,pre指针开始跑,
//当p指针跑到最后时,pre所指指针就是倒数第k个节点
while(p!=NULL){
p = p->next;
count++;
if(a < 1)
pre = pre->next;
a--;
}
//如果节点个数小于所求的倒数第k个节点,则返回空
if(count < k)
return NULL;
return pre;
}
};