题目:
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、数组法:我开始的思路是首先将所有节点都保存在一个数组中,然后在数组中删除掉倒数第n个元素即可。虽然这还是只用了一次遍历,但是空间复杂度却高达O(m),其中m是链表的长度(不过神奇的是,这种方法在Leetcode中的运行时间反而比双指针更短)。
2、双指针法:这是解决链表问题的一大法宝。定义p,q两个指针都指向头结点,然后先让p先走n步,接着让p和q同步前进直到p到达末节点。此时q所在的节点即为倒数第n个节点,删除即可。为了统一处理n为1的情况,我们可以设置一个虚拟头结点,让p,q从虚拟头结点开始前进,如下面代码中的pre_head所示。事实上在很多情况下,操作链表时如果添加一个虚拟头结点,可以让算法的框架更加简洁和明晰。
代码:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
if(head == NULL)
return NULL;
ListNode pre_head = ListNode(0); // the virtual head
pre_head.next = head;
ListNode *p = &pre_head, *q = &pre_head;
for(int i = 0; i < n; i++)
p = p->next;
while(p->next)
{
p = p->next;
q = q->next;
}
p = q->next;
q->next = p->next;
delete p;
return pre_head.next;
}
};