【剑指offer】链表相关-删除value值节点

#include<iostream.h>
#include <stdio.h>
#include <stack>

struct ListNode
{
int m_nValue;
ListNode * m_pNext;
};
ListNode * CreateLink(int a[],int k)
{
ListNode * Head=NULL,*q=NULL;
for(int i=0;i<k;i++)
{
ListNode * pNew=new ListNode();
pNew->m_nValue=a[i];
pNew->m_pNext=NULL;

if(Head==NULL)
{
Head=pNew;
q=pNew;
}
else
{
q->m_pNext=pNew;
q=q->m_pNext;
}
}
return Head;
}
//从头到尾打印列表
void printLink(ListNode * pHead)
{
cout<<"链表内容为:";
ListNode *p=pHead;
while(p)
{
cout<<p->m_nValue<<" ";
p=p->m_pNext;
}
cout<<endl;
}

//链表中找到第一个含有某值的节点，并删除该结点
void RemoveNode(ListNode ** pHead,int value)
{
if(pHead==NULL)
return;
if(*pHead==NULL)
return;
ListNode * p=*pHead,*q=NULL;
while(p)
{
if(p->m_nValue==value)
{
//第一个结点即为要删除的结点
if(q==NULL)
{
*pHead=(*pHead)->m_pNext;
delete p;
p=NULL;
return;
}
else
{
q->m_pNext=p->m_pNext;
delete p;
p=NULL;
return;
}
}
else
{
q=p;
p=p->m_pNext;
}
}

}
//书上的
void RemoveNode1(ListNode **pHead,int value)
{
if(pHead == NULL || *pHead ==NULL)
return;
ListNode * pToBeDeleted=NULL;
//特殊处理第一个结点即为要删除的结点,注意括号的使用
if((*pHead)->m_nValue == value)
{
pToBeDeleted=*pHead;
*pHead=(*pHead)->m_pNext;
}
else
{
ListNode * pNode=*pHead;
while(pNode->m_pNext!=NULL && pNode->m_pNext->m_nValue!=value)
pNode=pNode->m_pNext;
if(pNode->m_pNext!=NULL && pNode->m_pNext->m_nValue==value)
{
pToBeDeleted=pNode->m_pNext;
pNode->m_pNext=pNode->m_pNext->m_pNext;
}
}
if(pToBeDeleted !=NULL)
{
delete pToBeDeleted;
//这一步很重要，防止操作已经释放的内存空间
pToBeDeleted=NULL;
}
}
//======测试用例========
//1.没有包含该结点
void Test1()
{
cout<<"测试用例test1结果"<<endl;
int a[]={1,2,3};
ListNode * ptr=CreateLink(a,3);
printLink(ptr);
RemoveNode(&ptr,4);
printLink(ptr);
}
//2.结点在第一个位置
void Test2()
{
cout<<"测试用例test2结果"<<endl;
int a[]={1,2,3};
ListNode * ptr=CreateLink(a,3);
printLink(ptr);
RemoveNode(&ptr,1);
printLink(ptr);
}
//3.结点在最后一个位置
void Test3()
{
cout<<"测试用例test3结果"<<endl;
int a[]={1,2,3};
ListNode * ptr=CreateLink(a,3);
printLink(ptr);
RemoveNode(&ptr,3);
printLink(ptr);
}
//4.结点在中间
void Test4()
{
cout<<"测试用例test4结果"<<endl;
int a[]={1,2,3,4,5};
ListNode * ptr=CreateLink(a,5);
printLink(ptr);
RemoveNode(&ptr,3);
printLink(ptr);
}
//5.链表没有结点，空链表
void Test5()
{
cout<<"测试用例test5结果"<<endl;
ListNode * ptr=NULL;
printLink(ptr);
RemoveNode(&ptr,3);
printLink(ptr);
}
//6.只有一个结点
void Test6()
{
cout<<"测试用例test6结果"<<endl;
int a[]={1};
ListNode * ptr=CreateLink(a,1);
printLink(ptr);
RemoveNode(&ptr,1);
printLink(ptr);
}
void main()
{
Test1();
Test2();
Test3();
Test4();
Test5();
Test6();

}