描述: |
详细描述:
原型: ListNode* InsertNodeToList(ListNode* pListHead, ListNode* pInsertNode); 输入参数: ListNode* pListHead 单向链表 ListNode* pInsertNode 新插入节点 输出参数(指针指向的内存区域保证有效): ListNode* pListHead 单向链表 返回值: 正常插入节点返回链表头指针,其它异常返回空指针
|
#include "OJ.h"
/*
功能: 输入一个升序单向链表和一个链表节点,向单向链表中按升序插入这个节点。
输入为空指针的情况视为异常,另外不考虑节点值相等的情况。
输入: ListNode* pListHead 单向链表
ListNode* pInsertNode 新插入节点
输出: ListNode* pListHead 单向链表
返回: 正常插入节点返回链表头指针,其它异常返回空指针
*/
ListNode* InsertNodeToList(ListNode* pListHead, ListNode* pInsertNode)
{
/*在这里实现功能*/
if (pListHead == NULL || pInsertNode == NULL)
{
return (ListNode*)NULL;
}
ListNode *tmp = (ListNode*)NULL;
/*当只有一个节点的情况*/
if (pListHead->m_pNext == NULL)
{
if (pListHead->m_nKey < pInsertNode->m_nKey)
{
pListHead->m_pNext = pInsertNode;
pInsertNode->m_pNext = (ListNode*)NULL;
}
else
{
tmp = pListHead;
pListHead = pInsertNode;
pInsertNode->m_pNext = tmp;
}
return pListHead;
}
/*找到比当前节点大的第一个节点*/
ListNode *pre = pListHead;
ListNode *p = pre->m_pNext;
while (p)
{
if (p->m_nKey > pInsertNode->m_nKey)
{
break;
}
pre = p;
p = pre->m_pNext;
}
/*当没有找到,就把整个节点插入到尾节点*/
if (p == NULL)
{
pre->m_pNext = pInsertNode;
pInsertNode->m_pNext = (ListNode*)NULL;
}
else //插入到p和pre中间
{
pre->m_pNext = pInsertNode;
pInsertNode->m_pNext = p;
}
return pListHead;
}