题目
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。[1]
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
解题要点
快慢指针法:
- 初始快指针和慢指针都指向head,遍历链表,快指针跟着后移,并且保证快指针和慢指针之差在n+1内,直到快指针指向最后一个元素为止
- 特殊情况
- 删除头结点,则返回头结点下一个节点
- 删除的节点没有下一个节点
代码
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode slowNode = head;
int slowIndex = 0;
ListNode fastNode = head;
int fastIndex = 0;
int size = 1;
ListNode tempNode;
while ((tempNode = fastNode.next) != null) {
size++;
fastNode = tempNode;
fastIndex++;
if (fastIndex - slowIndex > n) {
slowNode = slowNode.next;
slowIndex++;
}
}
if (size == n) {
ListNode next = head.next;
head.next = null;
return next;
}
ListNode delNode = slowNode.next;
ListNode nextNode = delNode.next;
delNode.next = null;
slowNode.next = nextNode;
return head;
}
网友评论