[LeetCode]19. Remove Nth Node From End of List

Description

Given a linked list, remove the nth 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.

Try to do this in one pass.

Discussion

这个问题要求我们移除掉链表中的倒数第

n

个元素。常规思路当然是首先遍历一遍链表，得出链表的总长度

len

，再移除掉第

len-n

个元素。然而，题目的要求是一次遍历，这就要求我们有更加巧妙的方法。

我们可以设置两个指针

preLoc

和

curLoc

，使得两个指针相距为

n

，且一起往后运动。这样，当后者

curLoc

运动到链表尾部时，

preLoc

恰好在倒数第

n

的位置。

算法的时间复杂度为

O(n)

。

C++ Code

/**
* 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) {
ListNode * preLoc = head, * curLoc = head;      //设置两个指针相距n，当后面的指针curLoc运动到尾部时，前面的指针就在倒数第n个。
for(int i = 0; i < n; i++)
{
curLoc = curLoc->next;
}
//若恰好有n个元素
if(!curLoc)
{
return head->next;
}
while(curLoc->next)
{
curLoc = curLoc->next;
preLoc = preLoc->next;
}
preLoc->next = preLoc->next->next;
return head;
}
};