剑指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;
}