剑指Offer_面试题13_在O(1)时间删除链表结点

题目：给定单项链表的头指针和一个节点指针，定义一个函数在O(1)时间删除该结点。

分析：一般来说删除一个结点必须要知道他的前驱，这样才能让他的前驱的后继指向它的后继而“绕过”它，但这样时间复杂度就为O(n)了，必须得知道前驱。查前驱的时间复杂度为O(n), 但是查后继的复杂度为O(1), 可以把问题转换为删除后继，降低时间复杂度。

首先复制后继到自身，然后删除后继就可以了。但是注意链表的几种特殊情况：只有一个结点，要删除的是尾结点

#include <stdio.h>
#include <stdlib.h>
#include <stack>
#include <vector>
#include <iostream>
using namespace std;

//当需要获取目标节点的前驱时，如增加、删除节点等操作，判断条件通常是while(p->next != NULL)
//当需要获取目标节点时，如遍历节点、打印链表所有元素等操作，判断条件通常是while(p != NULL)

struct ListNode
{
int val;
ListNode* next;
};

//尾插法加入元素
void Add2Tail(ListNode** pHead, int value)
{
ListNode *pNew = new ListNode();
pNew->val = value;
pNew->next = NULL;

//空链表
if (*pHead == NULL)
{
*pHead = pNew;
}
else {
ListNode *pNode = *pHead;
while (pNode->next != NULL)
pNode = pNode->next;
pNode->next = pNew;
}
}

//打印链表元素
void PrintList(ListNode *head)
{
if (head == NULL)
return;
ListNode *p = head;
while (p != NULL) //这里如果是p->next != NULL则不会打印最后一个结点内容
{
std::cout << p->val << '\t';
p = p->next;
}
cout << endl;
}

//O(1)时间删除链表节点
void DeleteNode(ListNode* head, ListNode* node)
{
if (head == NULL || node == NULL) return;
if (node->next != NULL) //如果待删除的不是尾节点
{
ListNode *temp = node->next;
node->val = node->next->val;
node->next = node->next->next;
free(temp);
temp = NULL;
}
else if (node == head) //链表只有一个头结点，且删除头结点
{
free(node);
node = NULL;
head = NULL;
}
else //删除的是尾结点且链表不止一个结点
{
ListNode *temp = head;
while (temp->next != node)
temp = temp->next;
temp->next = NULL;
free(node);
node = NULL;
}
}

int main()
{
ListNode *head = NULL;
Add2Tail(&head, 67);
Add2Tail(&head, 0);
Add2Tail(&head, 24);
Add2Tail(&head, 58);
PrintList(head);

ListNode *rear = head;
while (rear->next != NULL)
rear = rear->next;
DeleteNode(head, rear); //删尾
PrintList(head);
DeleteNode(head, head); //删头
PrintList(head);
system("pause");
return 0;
}