题目:
输入一个链表,输出该链表中倒数第k个结点。
思路:
制造一把长度为 K 的尺子,尺子的后端走到链表末段,那么尺子的前端就是倒数第 K 个结点 。
代码:
package 链表中倒数第K个结点;
class ListNode {
int val;
ListNode next = null;
public ListNode(int val) {
this.val = val;
}
}
class Solution {
/**
* 制造一把长度为 K 的尺子,尺子的后端走到链表末段,那么尺子的前端就是倒数第 K 个结点
* @param head
* @param k
* @return
*/
public ListNode FindKthToTail(ListNode head, int k) {
ListNode pre, last;
pre = last = head; // 创建尺子的头尾结点并赋予头结点值
int i = 0;
for (; last != null; i++) { // 尾结点不为空时,尺子的尾结点向后移动,超出链表范围时,尾结点为空,头结点为倒数第 k 个结点
if (i >= k) // 当尺子的尾结点移动 k 个位置后,头尾结点开始同步向后移动
pre = pre.next;
last = last.next; // 尺子的尾结点持续移动,
}
return i < k ? null : pre; // 若 k 大于链表长度,返回空;小于链表长度,返回尺子的头结点
}
}
public class Demo {
public static void main(String[] args) {
ListNode listNode = new ListNode(1);
listNode.next = new ListNode(2);
listNode.next.next = new ListNode(3);
listNode.next.next.next = new ListNode(4);
listNode.next.next.next.next = new ListNode(5);
Solution solution = new Solution();
ListNode listNode2 = solution.FindKthToTail(listNode, 1);
System.out.println(listNode2.val);
}
}