Given a linked list, remove the nth node from the end of list and return its head.
For example,
Given linked list: 1->2->3->4->5, and n = 2. After removing the second node from the end, the linked list becomes 1->2->3->5.
Note:
Given n will always be valid.
Try to do this in one pass.
思路:
假设我们想要删除倒数第1个节点5,那么我们应该把指针停留在倒数第2个节点4处,因为我们要进行node.next = node.next.next的节点删除操作。由此可见,如果我们想要删除倒数第n个节点的话,我们应该把指针停留在倒数n+1个节点处。
我们需要两个指针,一个先走,一个后走。等前面走了n+1步之后,后面的指针再走,此时两个指针保持恒定的距离n+1一直行走,直到前面的指针达到链表尾部。而此时后面跟着的指针到链表尾部的距离为n+1,也就是处在倒数第n+1个节点处,它的next节点就是要删除的倒数第n个节点,接下来就可以进行节点删除操作。
注意:
- 一定要创建一个dummy节点让其next为head,因为有可能链表只有一个元素,删除之后就为空,处理起来就麻烦了。
- 当链表有环的时候,下面的程序不适用,因为会出现死循环。
public ListNode removeNthFromEnd(ListNode head, int n) {
if(head==null) return null;
ListNode dummy = new ListNode(0);
dummy.next = head;
ListNode h = dummy;
while(head != null) {
head = head.next;
if(n < 0 || --n < 0) { // n<0的话就不用再自减了,防止整数溢出
h = h.next;
}
}
h.next = h.next.next;
return dummy.next;
}
上面的代码还是不太好理解,下面的更好理解:
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode dummy = new ListNode(0);
dummy.next = head;
ListNode node = dummy;
while(n-->0) head = head.next;
while(head != null) {
head = head.next;
node = node.next;
}
node.next = node.next.next;
return dummy.next;
}
再补充个C++的代码:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode *node=head, **cur = &head;
while(n--) node = node->next;
while(node) {
node = node->next;
cur = &(*cur)->next;
}
*cur = (*cur)->next;
return head;
}