删除链表的节点（时间复杂度O(1)）

 /*题目：在O(1)时间内删除链表的节点(此答案基于一个假设：要删除的节点在此链表中，若需判断是否在，解整道题需O(N)的时间复杂度）
  给定单向链表的头指针和一个节点指针，定义一个函数在O(1)时间内删除该节点。*/
/*解题思想：把下一个节点的内容复制到需要删除的节点上覆盖原先的内容，再把下一个节点删除；
   分为三种情况：（1）要删除的节点不是尾结点；（2）要删除的是头节点，删除之后还要把头节点置为空；
                 （3）要删除的是尾结点，这时需要从头顺序遍历得到删除节点的前序节点，
				      并完成删除操作*/
 //第三种情况时间复杂度为O(n),而另两种情况时间复杂度为O(1),所以平均时间复杂度为[(n-1)*O(1)+O(n)]/n,结果还是o(1)。
#include<stdio.h>
#include<stdlib.h>
struct ListNode
{
	int data;
	ListNode *next;
};
void createListF(ListNode *&head, int a[], int n)
{
	ListNode *s;
	head = (ListNode*)malloc(sizeof(ListNode));
	head->next= NULL;
	int j = 0;
	for (j = 0; j < n; j++)
	{
		s = (ListNode*)malloc(sizeof(ListNode));
		s->data = a[j];
		s->next = head->next;
		head->next = s;
	}
}
void DeleteNode(ListNode *&head, ListNode *p)
{
	if (!head && !p)
	{
		return;
	}
	//要删除的节点不是尾结点
	if (p->next != NULL)
	{
		ListNode *q = p->next;
		p->data = q->data;
		p->next = q->next;
		delete q;
		q = NULL;
	}
	//要删除的是头节点
	else if (head == p)
	{
		delete p;
		p = NULL;
		head = NULL;
	}
	//要删除的节点是尾节点
	else
	{
		ListNode *s = head;
		while (s->next != p)
		{
			s = s->next;
		}
		s->next = NULL;
		delete p;
		p = NULL;
	}
}
void traverseList(ListNode *head)
{
	ListNode *p = head->next;
	while (p!= NULL)
	{
		printf("%d ", p->data);
		p = p->next;
	}
	printf("\n");
}
int main()
{
	int a[10] = { 2,31,3,21,5,4,2,7,8,9 };
	ListNode *head;
	createListF(head, a, 10);
	traverseList(head);
	ListNode *p = head->next->next->next;
	DeleteNode(head, p);
	traverseList(head);
	return 0;
}