题目:
实现一种算法,找出单向链表中倒数第 k 个节点。返回该节点的值。
示例:
输入: 1->2->3->4->5 和 k = 2
输出: 4
说明:
给定的 k 保证是有效的。
思路:
设置2个指针p和q都指向head,p先向前移动 k-1 个结点,接着,p,q一起移动,直到q.next = null,此时q指针所指的结点就是要找的结点。
例如下图(将每个格子看成链表):
k等于3时,初始时p和q都指向head,当p移动2个位置时,接下来q开始移动。
当p位于链表末尾时,q就是倒数第3个结点。
参考代码:
public int kthToLast(ListNode head, int k) {
/** 设置2个指针p,q,p从链表头走到链表的第k个位置,q开始从头往后走,
当p走到链表尾部时,q的位置就是倒数第k个节点
**/
if(head == null)return -1;
ListNode p = head;
ListNode q = head;
for(int i = 0;i < k - 1; k++) p = p.next;
while(p.next != null) {
p = p.next;
q = q.next;
}
return q.val;
}
注意:
for循环的终止条件容易写错,因为开始时p就指向head,相当于已经移动了1个单位,所以如果k = 3,则p循环2次,即到第3个结点时,结束for循环。
补充:
时隔3个月后,再来做题,发现牛可上面代码通不过,原因是:k值可能大于链表个数。
参考代码:
public ListNode FindKthToTail(ListNode head,int k) {
if(head == null || k < 0)return null;
ListNode p = head;
ListNode q = head;
int count = 1;
while(count != k){
p = p.next;
if(p == null)return null;//增加的代码,如果链表到末尾了,链表个数还没达到k,说明k有问题,返回null
count++;
}
while(p.next != null){
p = p.next;
q = q.next;
}
return q;
}