Leetcode19. 删除链表的倒数第N个节点
题目:
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
题解:
设定双指针 p 和 q,当 q 指向末尾的 NULL,p 与 q 之间相隔的元素个数为 n 时,那么删除掉 p 的下一个指针就完成了要求。
- 设置虚拟节点 dummyHead 指向 head
- 设定双指针 p 和 q,初始都指向虚拟节点 dummyHead
- 移动 p,直到 p 与 q 之间相隔的元素个数为 n
- 同时移动 p 与 q,直到 p 指向的为 NULL
- 将 q 的下一个节点指向下下个节点
双指针
java代码:
public static ListNode removeNthFromEnd(ListNode head, int n) {
ListNode dummy = new ListNode(-1);
dummy.next = head;
ListNode p = dummy;
ListNode q = dummy;
for (int i = 0; i <= n; i++) {
p = p.next;
}
while (p != null) {
p = p.next;
q = q.next;
}
q.next = q.next.next;
return dummy.next;
}
scala代码:
/**
* 双指针
*
* @param head
* @param n
* @return
*/
def removeNthFromEnd(head: ListNode, n: Int): ListNode = {
val dummy = new ListNode(0)
dummy.next = head
var p = dummy
var q = dummy
for (i <- 0 to n) {
p = p.next
}
while (p != null) {
p = p.next
q = q.next
}
q.next = q.next.next
dummy.next
}