题目描述
输入一个长度为n的链表,设链表中的元素的值为ai ,输出一个链表,该输出链表包含原链表中从倒数第k个结点至尾节点的全部节点。(如果该链表长度小于k,请返回一个长度为 0 的链表)
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
第一种解法:
思路: 先循环得到链表的长度,再做判断
struct ListNode* getKthFromEnd(struct ListNode* head, int k){
int i = 0,j;
struct ListNode* p = head;
if(head == NULL) {
return NULL;
}
while(p != NULL) {
p = p->next;
i++;
}
p = head;
if(i < k) {
return NULL;
}
for(j = 0;j < i - k;j++) {
p = p->next;
}
return p;
}
第一种解法简单粗暴,效率低,我们再介绍第二种解法
第二种解法:
思路: 第二种解法是采用双指针,一个快指针,一个慢指针。先让快指针走k步,指到第k个结点,再让慢指针和快指针一起走,当快指针指到尾部时,慢指针刚好指到倒数第k个结点。
struct ListNode* getKthFromEnd(struct ListNode* head, int k){
struct ListNode* pfast = head;
struct ListNode* pslow = head;
if(head == NULL || k <= 0) {
return NULL;
}
k = k - 1;
while(k--) {
if(pfast->next != NULL) {
pfast =pfast->next;
}else {
return NULL;
}
}
while(pfast->next != NULL) {
pfast = pfast->next;
pslow = pslow->next;
}
return pslow;
}
两种解法的对比: 暴力循环解法花费4ms,双指针耗时0ms 题目及执行时长数据来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof
|