题目
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
struct ListNode {
int val;
ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
思路
1.构造双指针
首先我们将添加一个哑结点作为辅助,该结点位于列表头部。哑结点用来简化某些极端情况,例如列表中只含有一个结点,或需要删除列表的头部。在第一次遍历中,我们找出列表的长度 LL。然后设置一个指向哑结点的指针,并移动它遍历列表,直至它到达第 (L - n)(L−n) 个结点那里。我们把第 (L - n)(L−n) 个结点的 next 指针重新链接至第 (L - n + 2)(L−n+2) 个结点,完成这个算法。
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* copyList = head;
int lengthOfList = 0;
while (copyList != NULL)
{
lengthOfList++;
copyList = copyList->next;
}
lengthOfList -= n;
ListNode* result = copyList;
while (lengthOfList > 0)
{
lengthOfList--;
copyList = copyList->next;
}
copyList->next = copyList->next->next;
return result->next;
}
2.构造双指针(两个指针间隔n)
上述算法可以优化为只使用一次遍历。我们可以使用两个指针而不是一个指针。第一个指针从列表的开头向前移动 n+1步,而第二个指针将从列表的开头出发。现在,这两个指针被 n 个结点分开。我们通过同时移动两个指针向前来保持这个恒定的间隔,直到第一个指针到达最后一个结点。此时第二个指针将指向从最后一个结点数起的第 n个结点。我们重新链接第二个指针所引用的结点的 next 指针指向该结点的下下个结点。
ListNode* removeNthFromEnd2(ListNode* head, int n)
{
ListNode* preNode = head;
ListNode* curNode = head;
while (n-- != 0) curNode = curNode->next;
if (curNode == NULL) return preNode;
while (curNode->next != NULL)
{
preNode = preNode->next;
curNode = curNode->next;
}
preNode->next = preNode->next->next;
return head;
}
/**插入指针**/
void inserNode(ListNode* p, int i)
{
ListNode* node = new ListNode(1);
node->val = i;
node->next = p->next;
p->next = node;
}
int main(int argc, char* argv[])
{
int a[] = { 1,2,3,4,5 };
ListNode* test = new ListNode(1);
ListNode* result;
for (int i : a)
{
Solution().inserNode(test, i);
}
result = Solution().removeNthFromEnd(test, 2);
system("pause");
return 0;
}