【Java】 剑指offer(22) 链表中倒数第k个结点

时间:2023-12-20 17:02:44

正文

本文参考自《剑指offer》一书,代码采用Java语言。

更多:《剑指Offer》Java实现合集  

题目

  输入一个链表,输出该链表中倒数第k个结点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾结点是倒数第1个结点。例如一个链表有6个结点,从头结点开始它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个结点是值为4的结点。

思路

  第一直觉是先从头开始遍历,计算链表个数n,然后重新遍历,第n-k+1个结点即为所需要的结点。但是需要遍历2次。后面采用了栈进行实现该方法,空间复杂度比较大。书中的方法则是:设置两个指针,第一个指针先遍历k-1步;从第k步开始,第二个指针指向头结点,两个结点同时往后遍历,当第一个指针到达最后一个结点时,第二个指针指向的正好是倒数第k个结点。(其实感觉还是近似遍历了两次啊。。)

测试算例 

  1.功能测试(第k个结点分别在链表的中间,头结点和尾结点)

  2.特殊测试(头结点为null,k超出范围)

完整Java代码

(含测试代码)

package _22;

import java.util.Stack;

/**
*
* @Description 面试题22:链表中倒数第k个结点
*
* @author yongh
* @date 2018年10月14日 下午4:23:32
*/ //题目:输入一个链表,输出该链表中倒数第k个结点。为了符合大多数人的习惯,
//本题从1开始计数,即链表的尾结点是倒数第1个结点。例如一个链表有6个结点,
//从头结点开始它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个结点是
//值为4的结点。 public class KthNodeFromEnd {
public static class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
} } //方法1:利用栈
public ListNode FindKthToTail1(ListNode head,int k) {
if(head==null || k<=0)
return null;
int numbOfList=1;
Stack<ListNode> st = new Stack<ListNode>();
st.push(head);
ListNode node=head.next;
while(node!=null){
numbOfList++;
st.push(node);
node=node.next;
}
if(k>numbOfList)
return null;
else{
for(int i=1;i<=k;i++){
node=st.pop();
}
return node;
}
} //方法2:利用两个相距为k-1的指针
public ListNode FindKthToTail2(ListNode head,int k) {
if(head==null || k<=0)
return null;
ListNode pAhead=head;
for(int i=1;i<k;i++){
pAhead=pAhead.next;
if(pAhead==null)
return null;
}
ListNode pBehind = head;
while(pAhead.next!=null) {
pAhead=pAhead.next;
pBehind=pBehind.next;
}
return pBehind;
} //=====================测试代码======================= /*
* null
*/
void test1() {
ListNode head=null;
ListNode result=FindKthToTail2(head, 1);
if(result==null)
System.out.println("test1 passed!");
else
System.out.println("test1 failed!");
} /*
* k超出范围
*/
void test2() {
ListNode head=new ListNode(2);
ListNode result=FindKthToTail2(head, 2);
if(result==null)
System.out.println("test2 passed!");
else
System.out.println("test2 failed!");
} /*
* 单个结点
*/
void test3() {
ListNode head=new ListNode(1);
ListNode result=FindKthToTail2(head, 1);
if(result.val==1)
System.out.println("test3 passed!");
else
System.out.println("test3 failed!");
} /*
* 尾结点
*/
void test4() {
ListNode node1=new ListNode(1);
ListNode node2=new ListNode(2);
ListNode node3=new ListNode(3);
ListNode node4=new ListNode(4);
node1.next=node2;
node2.next=node3;
node3.next=node4;
ListNode result=FindKthToTail2(node1, 1);
if(result.val==4)
System.out.println("test4 passed!");
else
System.out.println("test4 failed!");
} /*
* 中间结点
*/
void test5() {
ListNode node1=new ListNode(1);
ListNode node2=new ListNode(2);
ListNode node3=new ListNode(3);
ListNode node4=new ListNode(4);
node1.next=node2;
node2.next=node3;
node3.next=node4;
ListNode result=FindKthToTail2(node1, 2);
if(result.val==3)
System.out.println("test5 passed!");
else
System.out.println("test5 failed!");
} /*
* 头结点
*/
void test6() {
ListNode node1=new ListNode(1);
ListNode node2=new ListNode(2);
ListNode node3=new ListNode(3);
ListNode node4=new ListNode(4);
node1.next=node2;
node2.next=node3;
node3.next=node4;
ListNode result=FindKthToTail2(node1, 4);
if(result.val==1)
System.out.println("test6 passed!");
else
System.out.println("test6 failed!");
} public static void main(String[] args) {
KthNodeFromEnd demo = new KthNodeFromEnd();
demo.test1();
demo.test2();
demo.test3();
demo.test4();
demo.test5();
demo.test6();
}
}

  

test1 passed!
test2 passed!
test3 passed!
test4 passed!
test5 passed!
test6 passed!

KthNodeFromEnd

收获

  1.注意代码的鲁棒性,开始思考前都需要注意特殊输入测试;

  2.一个指针遍历链表无法解决问题时,可以考虑使用两个指针来遍历链表:两个指针先后遍历(即该题目)、或者两个指针遍历速度不同(如:求链表中的中间结点,可以令一个指针一次走一步,另一个指针一次走两步来实现)

  

更多:《剑指Offer》Java实现合集