题目:
如果给定一个链表,请问如何删除链表中的倒数第k个节点?假设链表中节点的总数为n,那么1<=k<=n。要求只能遍历链表一次。例如,输入下图a中的链表,删除倒数第2个节点之后的链表如图b。
分析:
该题如果不要求一次遍历,可以用一个指针遍历两次链表,一次遍历查链表中元素个数n,第二次遍历链表时可以找出链表中第n-k个节点也就是倒数第k+1个节点,然后把倒数第k+1个节点的next指针指向倒数第k-1个节点就能完成删除倒数第k个节点的任务了。
但要求只能遍历一次,因此我们可以采取双指针策略,为了能顺利找到倒数第k+1个节点,第一个指针p1从链表头结点开始走k步,走到第k+1步时p2指针从链表头节点开始和指针p1保持相同速度遍历链表,由于两者距离始终是k,当指针p1指向尾结点时,p2正好指向倒数第k+1个节点。
下图就是删除倒数第2个节点的例子。
public class RemoveNthFromEnd {
public static void main(String[] args) {
ListNode l1 = new ListNode(1);
ListNode l2 = new ListNode(2);
ListNode l3 = new ListNode(3);
ListNode l4 = new ListNode(4);
ListNode l5 = new ListNode(5);
l1.next = l2;
l2.next = l3;
l3.next = l4;
l4.next = l5;
ListNode listNode = removeNthFromEnd(l1, 2);
System.out.println(listNode);
}
public static ListNode removeNthFromEnd(ListNode head,int n){
//运用哨兵节点来避免单独处理删除头节点的情况
ListNode dummy = new ListNode(0);
dummy.next = head;
ListNode front = head,back = dummy;
for (int i = 0; i <n; i++) {
front = front.next;
}
while (front != null){
back = back.next;
front = front.next;
}
back.next = back.next.next;
return dummy.next;
}
}