剑指Offer——面试题18（一）：在O(1)时间删除链表结点

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

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;
struct ListNode{
	int value;
	ListNode* next;
}; 
/**
 * 题目要求时间复杂度为O(1), 我们不能用遍历的思想对结点进行删除,
 *  我们可以通过将要删除结点next的值赋值到要删除的结点,再让该结点指向next的next结点即可 
 *   特殊情况：1、要删除的结点为最后一个结点, 则需要从头开始遍历进行删除
 *             2、链表只有一个结点,删除该结点后,需要将链表的头结点设置为NULL 
 *    对于尾节点而言, 由于仍然需要顺序查找, 时间复杂度是O(n), 因此,总的平均时间复杂度是
 *    [(n-1)xO(1)+O(n)]/n, 结果还是 O(1) 
 **/
void deleteNode(ListNode** pListHead, ListNode* pToDeleted) {
	if(!pListHead || !pToDeleted) return ;
	
	// 1、要删除的结点不是尾节点
	if(pToDeleted->next!=NULL){
		ListNode* pNext = pToDeleted->next;
		pToDeleted->value=pNext->value;
		pToDeleted->next=pNext->next;
		delete pNext;
		pNext=NULL;
	}
	// 2、链表只有一个结点
	else if(*ListNode==pToDeleted){
		delete pToDeleted;
		pToDeleted=NULL;
		*pListHead=NULL;
	} 
	// 3、链表中有多个结点,删除尾节点
	else {
		ListNode* pNode = *pListHead;
		while(pNode->next!=pToDeleted){
			pNode=pNode->next;
		}
		pNode->next=NULL;
		delete pToDeleted;
		pToDeleted=NULL;
	}
} 
int main() {
	return 0;
}