LeetCode: Remove Nth Node From End of List

题目

https://oj.leetcode.com/problems/remove-nth-node-from-end-of-list/

分析

1. 两个指针， 第一个指针先走n步， 然后两个指针一起走， 直到第一个指针走到结尾处， 此时第二个指针即为要删除的结点的上一个结点。

2. 特别要注意的一点是， 如果要删除的结点是第一个结点， 要特别的处理一下， 因为第二个指针无法指向第一个结点的上一个结点。

代码

class Solution
{
public:
ListNode *removeNthFromEnd(ListNode *head, int n)
{
ListNode *l, *r;

l = head;
r = head;
for (int i = 0; i < n; i++)
r = r->next;

if (r == NULL)
{
ListNode *t = head;
head = head->next;
delete t;
}
else
{
while (r->next != NULL)
{
l = l->next;
r = r->next;
}
ListNode *t = l->next;
l->next = t->next;
delete(t);
}
return head;
}
};