剑指 Offer 22. 链表中倒数第k个节点
题目
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。
例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
题解
双指针算法,只需要遍历一次链表。可以先让第一个指针保持不动,第二个指针向后走k步;然后两个指针一同向后走,直到判断条件为false(也可以第二个指针先向后走k - 1 步, 但注意判断条件要发生变化)。在本题中需要注意处理head为空,k <= 0,k大于链表长度的情况。前两个比较容易想出,但第三个会是经常遗漏的点。
Code
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode getKthFromEnd(ListNode head, int k) {
if(head == null || k <= 0){
return null;
}
ListNode p = head, q = head;
while(k-- > 0){
// 如果k大于链表长度
if(q == null){
return null;
}else{
q = q.next;
}
}
while(q != null){
p = p.next;
q = q.next;
}
return p;
}
}
面试题23
题目:如果一个链表中包含环,如何找出环的入口节点?
Code:
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode findEntryNodeOfLoop(ListNode head) {
if(head == null){
return null;
}
// 先定义两个指针, 一个指针一次走一个节点,一个指针一次走两个节点
ListNode p = head, q = head;
while(q != null){
// 不存在环时p会走到null
if(p.val == q.val){
// q追上了p,存在环
break;
}
p = p.next;
q = q.next.next;
}
if(q == null){
// 不存在环,直接退出
return null;
}
// 从相遇节点开始,计算环中节点个数,环中至少存在一个节点
int count = 1;
// 从下一个节点开始判断,特判环中只有一个节点,count从1开始计数
q = p.next;
while(q.val != p.val){
count++;
}
// 先让第一个指针向前走count个节点,然后两个指针以相同的速度向后走,相遇的节点即为入口节点
// 与倒数第k的节点的例子相似,入口节点即为倒数count个节点,如果按上面的方式,最后第一个节点
// 不会为null,而是指向下一个节点——入口节点,第二个节点指向倒数第count个节点,也是入口节点
q = head;
p = head;
while(count-- > 0){
q = q.next;
}
while(q != null){
p = p.next;
q = q.next;
}
return p;
}
}