题目:
输入一个链表,输出该链表中倒数第k个结点。
分析:
链表只能从链表头结点向后一个个访问,要求倒数第K个,需要知道一共有多少个,进而知道倒数第K个是正数多少个。求个数需要先遍历一遍。
最先想到的方式往往不是最优的,可以进一步改进。
链表的题目要结合两个指针快慢走操作,两个指针一次遍历可以找到倒数第K个,可以先让一个走K-1步,一个不动,此时两个相差K步,之后两个指针同时走,先走的指针到达尾部时,慢走指针到达要求的结点。
实现:
public ListNode FindKthToTail(ListNode head, int k) {
// 保证代码鲁棒性
if (head == null || k <= 0)
return null;
ListNode fast = head;
ListNode slow = head;
// 倒数第K个,是K-1步
for (int i = 0; i < k - 1; i++) {
if (fast.next == null) {
return null;
}
fast = fast.next;
}
while (fast.next != null) {
slow = slow.next;
fast = fast.next;
}
return slow;
}
其中结点定义如下:
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}