向升序单向链表中插入一个节点

#include "OJ.h"

/*
功能: 输入一个升序单向链表和一个链表节点，向单向链表中按升序插入这个节点。
输入为空指针的情况视为异常，另外不考虑节点值相等的情况。

输入: ListNode* pListHead  单向链表
ListNode* pInsertNode 新插入节点

输出: ListNode* pListHead  单向链表

返回: 正常插入节点返回链表头指针，其它异常返回空指针
*/
ListNode* InsertNodeToList(ListNode* pListHead, ListNode* pInsertNode)
{
/*在这里实现功能*/

if ((ListNode *)NULL == pListHead )
{
return (ListNode *)(NULL);
}
if ((ListNode *)NULL == pInsertNode )
{
return (ListNode *)(NULL);
}
if (pInsertNode->m_nKey < pListHead->m_nKey)
{
pInsertNode->m_pNext = pListHead;
pListHead = pInsertNode;
return (ListNode *)pListHead;
}
ListNode* p = pListHead;

while ((ListNode *)NULL != p->m_pNext && p->m_pNext->m_nKey < pInsertNode->m_nKey)
{
p = p->m_pNext;
}

pInsertNode->m_pNext = p->m_pNext;
p->m_pNext = pInsertNode;
return (ListNode *)pListHead;
}

int main()
{
ListNode astListNode[2];
ListNode stInsertNode;
ListNode *pListHead;

astListNode[0].m_nKey = 2;
astListNode[0].m_pNext = &astListNode[1];

astListNode[1].m_nKey = 3;
astListNode[1].m_pNext =(ListNode *)(NULL);

stInsertNode.m_nKey = 4;
stInsertNode.m_pNext =(ListNode *)(NULL);

pListHead = InsertNodeToList(astListNode, &stInsertNode);
return 0;
}