链表中倒数第k个结点
描述
输入一个链表,输出该链表中倒数第k个结点。
代码 (Java)
/*
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}*/
public class Solution {
public ListNode FindKthToTail(ListNode head,int k) {
if (head == null || k == 0)
return null;
ListNode pFirst = head, pAfter = head;
// pFirst需要领先pAfter (k-1)个位置
for (int i = 0; i < k - 1; i++) {
// 需要处理k大于总长度的情况
if (pFirst.next != null) {
pFirst = pFirst.next;
} else
return null;
}
while (pFirst.next != null) {
pFirst = pFirst.next;
pAfter = pAfter.next;
}
return pAfter;
}
}
思路
- 本题代码时间复杂度是O(n),用到了两个指针,让第一个指针领先第二个指针(k-1)个位置进行遍历,这样当第一个指针到达尾部时,第二个指针就是倒数第k个节点;
- 需要注意处理k大于总长度的情况。