题目:
输入一个链表,输出链表中倒数第K个结点。
答案:
常规思路就是要经过两次遍历,不够高效。
先遍历一遍链表,得到链表的长度n,再减去k,得到结点的正数位置,最后再遍历得到这个结点。
双指针法。只需要遍历一次链表
第一个指针p1先走k步
然后第二个指针p2与p1同时走,这样p1和p2之间一直保持k个结点的距离。
因此直到p1走到最后一个结点,此时p2正好位于倒数第k个结点。
public ListNode getLastK(ListNode head, int k) {
if (head == null || k <= 0) {
throw new RuntimeException("非法输入");
}
ListNode p1 = head;
ListNode p2 = head;
//p1先走k步
for (int i = 0; i < k; i++) {
if (p1.next != null) {
p1 = p1.next;
} else {
return null;
}
}
//p1和p2同时走,直到p1走到最后结点,此时p2刚好指向倒数第k个结点
while (p1.next != null) {
p1 = p1.next;
p2 = p2.next;
}
return p2;
}