题目描述
输入一个链表,输出该链表中倒数第k个结点。
解题思路
经典的双指针法。定义两个指针,第一个指针从链表的头指针开始遍历向前走k-1步,第二个指针保持不动,从第k步开始,第二个指针也开始从链表的头指针开始遍历,由于两个指针的距离保持在k-1,当第一个指针到达链表的尾节点时,第二个指针刚好指向倒数第k个节点。
关注要点
1)链表头指针是否为空,若为空则直接返回回null
2)k是否为0,k为0也就是要查找倒数第0个节点,由于计数一般是从1开始的,所有输入0没有实际意义,返回null
3)k是否超出链表的长度,如果链表的节点个数少于k,则在指针后移的过程中会出现next指向空指针的错误,所以程序中要加一个判断
参考代码(Java)
- 结构定义
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}
- 解法一代码
public class Demo03 {
public ListNode FindKthToTail(ListNode head,int k) {
if(head==null || k==0) {
return null;
}
ListNode p1 = head;
//判断k是否超过链表节点的个数,注意是 i 从1~k-1
for (int i = 1; i < k; i++) {
if(p1.next!=null) {
p1=p1.next;
}else {
return null;
}
}
ListNode p2 = head;
while(p1.next!=null) {
p1=p1.next;
p2=p2.next;
}
return p2;
}
}
- 测试主方法
public static void main(String[] args) {
ListNode listNode = new ListNode(1);
ListNode listNode2 = new ListNode(2);
ListNode listNode3 = new ListNode(3);
ListNode listNode4 = new ListNode(4);
listNode3.next = listNode4;
listNode2.next = listNode3;
listNode.next = listNode2;
ListNode tail = FindKthToTail(listNode,2);
System.out.println(tail.val);
}
- 输出结果
3
- 解法二代码(用栈)
public class Demo03 {
public ListNode FindKthToTail(ListNode head,int k) {
if (head == null || k == 0) {
return null;
}
// 可以先把链表反转,然后找出第k个
Stack<ListNode> stack = new Stack<ListNode>();
int count = 0;
while (head != null) {
stack.push(head);
head = head.next;
count++;
}
if (count < k) {
return null;
}
ListNode knode = null;
for (int i = 0; i < k; i++) {
knode = stack.pop();
}
return knode;
}
}