原题如下: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.
题目的意思是给定一个链表,然后给定一个数值N,然后删除倒数第N 个元素,给定的N是合法的。在一次遍历中完成操作。
解题思路:先让一个指针走找到第N个节点,然后再让一个指针指向头结点(头节点不算是第一个节点),然后两具指针一起走,这时候第一个指针永远比第二个指针快N步,直到前一个指针直到了末尾,后一个指针就是倒数第N+1个结点,即这时候第一个指针为末尾最后一个指针,而第二个指针指向的是倒数的第N+1个指针,删除倒数第N个结点就可以了。
代码如下:
struct ListNode
{
int val;
ListNode * next;
ListNode(int x) : val(x), next(NULL) {}
};
ListNode* removeNthFromEnd(ListNode* head, int n)
{
ListNode * pa = head;
ListNode * pb = head;
for (int i = 0; i < n &&pa!=NULL; i++)
{
pa = pa->next;
}
if (pa=NULL)
{
head = head->next;
return head;
}
while (pa->next!=NULL)
{
pa = pa->next;
pb = pb->next;
}
pb->next = pb->next->next;
return head;
}