Given a linked list, remove the nth node from the end of list and return its head.
这道题的要求是删除链表的倒数第k个节点, 只要能查找到倒数第k个节点,删除啥的就好说了。
实现查找不难,两次遍历链表,第一遍记录链表长度,第二次直接遍历到倒数第k个即可。难点在于如何设计高效的算法一次遍历链表实现查找。
思路: 快慢指针。能否这样想,倒数第k个就是从末尾往前数第k个,而最后一个节点是空指针。设计两个指针,如果两个指针的距离始终是k,那么当后一个指针到尾节点时,前一个指针恰恰指向倒数第k个节点。这样,大概思路就出来了,后指针先遍历k个位置,然后前、后指针同步遍历,当后指针为空时,就找到了倒数第k个位置的节点。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
public class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode p=head;//快(后)指针
ListNode pre=head;//前指针的前指针,删除节点q使用
ListNode q=head.next;//慢(前)指针
for(int i=0;i<n;i++)
p=p.next;
if (p == null) //考虑边界情况,删除头结点
{return pre.next;
//System.out.println("error n");
}
else if(p.next==null){//考虑边界情况,删除尾节点
pre.next=q.next;
return pre;
}
while(p.next!=null){
p=p.next;
q=q.next;
pre=pre.next;
}
pre.next=q.next;
return head;
}
}
结果:
链表:12345678
K:3
输出:6