题目
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
算法1
public class P19_RemoveNode {
public ListNode removeNthFromEnd(ListNode head, int n) {
if(head == null || head.next == null){
return null;
}
int length = 1;
ListNode p = head.next;
while(p!=null){
length++;
p = p.next;
}
int index = length - n + 1;
if(index == 1){
return head.next;
}
ListNode blankHead = new ListNode(0);
blankHead.next = head;
p = head;
int i = 1;
while(i<index-1){
i++;
p = p.next;
}
p.next = p.next.next;
return blankHead.next;
}
}
思路:这里用到了最简单的两次遍历算法,第一次遍历确定长度,第二次遍历删除某位置节点,但是算法耗时出乎意料的少。
算法2(来自leetcode)
public class P19_RemoveNode2 {
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode dummy = new ListNode(0);
dummy.next = head;
ListNode first = dummy;
ListNode second = dummy;
// Advances first pointer so that the gap between first and second is n nodes apart
for (int i = 1; i <= n + 1; i++) {
first = first.next;
}
// Move first to the end, maintaining the gap
while (first != null) {
first = first.next;
second = second.next;
}
second.next = second.next.next;
return dummy.next;
}
}
思路:
上述算法可以优化为只使用一次遍历。我们可以使用两个指针而不是一个指针。第一个指针从列表的开头向前移动 n+1 步,而第二个指针将从列表的开头出发。现在,这两个指针被 n 个结点分开。我们通过同时移动两个指针向前来保持这个恒定的间隔,直到第一个指针到达最后一个结点。此时第二个指针将指向从最后一个结点数起的第 n 个结点。我们重新链接第二个指针所引用的结点的 next 指针指向该结点的下下个结点。