题目:输入一个链表,输出该链表中倒数第k个结点。
链表节点定义如下:
struct ListNode
{
int value;
ListNode* next;
}
思路:
定义两个指针,第一个指针先指向第k个节点,既向前走k-1步,然后另第二个指针指向头结点,然后两个指针一起向后移动,直到第一个指针走到最后一个节点时,此时第二个指针正好指向倒数第k个节点。
注意:
1.要判断输入是否为空指针,若为空指针,代码试图访问空指针会造成程序崩溃,所以此时要返回nullptr。
2.若k大于链表的节点数,此时也会试图访问空指针,也要返回nullptr。
3.k==0时,k-1是无符号数,不是-1,也会造成崩溃。
4.先让快指针走 k-1 步,并且当最后快指针指向最后一个元素(并非指向null)时停止。
基于以上思路,java参考代码如下:
package chapter3;
public class P134_KthNodeFromEnd {
public static class ListNode{
int val;
ListNode next;
public ListNode(int val){
this.val=val;
this.next=null;
}
}
public static ListNode findKthToTail(ListNode head,int k){
if(head==null||k<=0)
return null;
ListNode cur=head,fast=head;
for(int i=1;i<k;++i){//注意点1:因为最后快指针停在倒数第一个节点处,所以是i<k.
fast=fast.next;
if(fast==null)//判断是否小于k
return null;
}
while(fast.next!=null){//注意点2:让快指针最后停在最后一个节点处
cur = cur.next;
fast = fast.next;
}
return cur;
}
public static void main(String[] args){
ListNode head = new ListNode(1);
head.next= new ListNode(2);
head.next.next = new ListNode(3);
System.out.println(findKthToTail(head,1).val);//3
System.out.println(findKthToTail(head,2).val);//2
System.out.println(findKthToTail(head,3).val);//1
System.out.println(findKthToTail(head,4));//null
}
}
注意点见代码注解处。
相关题目:求链表的中间节点
同样需要用两个指针,同时从链表的头节点出发,一个指针一次走一步,另一个指针一次走两步。快的到末尾,慢的刚好在中间。
java参考代码如下:
package chapter3;
public class P138_findMiddleNode {
public static class ListNode{
int val;
ListNode next;
public ListNode(int val){
this.val=val;
this.next=null;
}
}
public static ListNode findmiddlenode(ListNode head){
if(head==null||head.next==null) return null;
ListNode cur=head,fast=head;
while (fast!=null&&fast.next!=null){
cur=cur.next;
fast=fast.next.next;
}
return cur;
}
public static void main(String[] args){
ListNode head = new ListNode(1);
head.next= new ListNode(2);
head.next.next = new ListNode(3);
head.next.next.next = new ListNode(4);
System.out.println(findmiddlenode(head).val);//2
}
}
C++参考代码如下:
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k)
{
if (pListHead==nullptr||k==0)
return nullptr;
ListNode* p1= pListHead;
ListNode* p2= nullptr;
for(unsigned int i=0;i<k-1;++i)
{
if( p1->next!= nullptr)
{
p1=p1->next;
}
else //如果链表的节点数少于k,返回一个nullptr
{
return nullptr;
}
}
p2=pListHead;
while (p1->next!=nullptr) //直到p1指向链表末尾
{
p1=p1->next;
p2=p2->next;
}
return p2;
}
测试用例:
a.功能测试(第k个节点在链表的中间;第k个节点是链表的头节点;第k个节点是链表的尾节点)。
b.特殊输入测试(链表头节点为nullptr指针;链表的节点总数少于k;k等于0)。
参考:
https://blog.csdn.net/qq_42209189/article/details/80353836
https://www.jianshu.com/p/0a2b3b783a54