题目描述
Given a linked list, remove the n-th node from the end of list and return its head.
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.
Follow up:
Could you do this in one pass?
分析
有关链表的经典题目,删除链表的倒数第二个元素,要求只遍历一遍。一个直观的想法就是先遍历一遍,求出链表一共有多少个元素,然后第二遍遍历的时候就可以直接定位到需要删除的元素。但是这道题要求只能遍历一遍,我们可以这样思考,正数第k个元素到头部的距离,和倒数第k个元素到尾部的距离是相同的。那么就可以使用两个指针,分别指向头部和正数第k个元素,然后开始遍历链表。当前面的指针已经指向链表尾部时,后面的指针所指向的就是倒数第k个元素,将其删掉即可。在具体实现时,需要将指针指向待删除元素的前面一个元素,这点一定要注意。
AC代码如下:
/**
* 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 head;
}
ListNode* temp1 = head;
ListNode* temp2 = head;
//前面的指针指向正数第n个元素的后一位
//因为需要使后面的指针在遍历结束时指向待删除元素的前一位
while(n--)
{
temp1 = temp1->next;
}
if(temp1 == NULL)//删除第二个元素
{
return head->next;
}
else
{
while(temp1->next != NULL)//是否遍历到尾部
{
temp1 = temp1->next;
temp2 = temp2->next;
}
temp2->next = temp2->next->next;//删除元素
}
return head;
}
};