题目:输入一个链表,输出该链表中倒数第k个节点。链表的定义如下:
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
思路:首先要注意代码的鲁棒性:当输入链表节点的数量小于k时,应该返回null。定义两个指针,第一个指针先从头节点开始沿链表前进k-1个节点到达第k个节点,此时令第二个指针指向头节点。然后让一前一后两个指针同时向后移动,当第一个指针到达尾节点时,第二个指针指向的便是倒数第k个节点。
代码(已在牛客网AC):
public ListNode FindKthToTail(ListNode head,int k) {
if(head == null || k < 1)
return null;
ListNode pAhead = head, pBehind = null;
// pAhead移动过程中检查该链表节点数是否小于k
for(int i = 0; i < k-1 ; i++)
{
if(pAhead.next != null)
pAhead = pAhead.next;
// 若链表节点数小于k则返回null
else
return null;
}
pBehind = head;
while(pAhead.next != null)
{
pAhead = pAhead.next;
pBehind = pBehind.next;
}
return pBehind;
}
性能:此时只需要遍历一次链表,故时间复杂度为O(n)。