题目:输入两个递增排序的链表,合并这两个链表并使新链表中的结点仍然是按照递增排序的。链表结点定义如下:
struct ListNode
{
int value;
ListNode *pNext;
};
思路:注意鲁棒性。
代码:
ListNode* Merge(ListNode* pHead1, ListNode* pHead2)
{
if(pHead1 == NULL)
return pHead2;
else if(pHead2 == NULL)
return pHead1;
ListNode* pMergedHead = NULL;
if(pHead1->m_nValue < pHead2->m_nValue)
{
pMergedHead = pHead1;
pMergedHead->m_pNext = Merge(pHead1->m_pNext, pHead2);
}
else
{
pMergedHead = pHead2;
pMergedHead->m_pNext = Merge(pHead1, pHead2->m_pNext);
}
return pMergedHead;
}
另一种思路,不用递归的方法,递归如果链表过长,可能会造成堆栈溢出。
用另一个新的链表表示合并后的链表,依次比较两个链表,每次把两个链表中value小的结点插入到新的链表中。代码如下(代码中没有创建新的结点来插入到新的链表中,只是把原链表中的结点取出来了):
ListNode* Merge(ListNode* pHead1, ListNode* pHead2)
{
if(pHead1 == NULL)
return pHead2;
else if(pHead2 == NULL)
return pHead1;
ListNode *first = pHead1;
ListNode *second = pHead2;
ListNode *result = NULL;
/// 先创建返回的结果,result表示新的链表的头结点
if(first->val <= second->val)
{
result = first;
first = first->next;
}
else
{
result = second;
second = second->next;
}
/// 创建一个新的指针指向新链表的当前索引
ListNode *node = result;
while(first != NULL && second != NULL) /// 两个链表中只要有一个链表指向空了就结束循环
{
if(first->val <= second->val)
{
node->next = first; /// 插入到新的链表后
first = first->next;
}
else
{
node->next = second;
second = second->next;
}
if(node->next != NULL) /// 如果不为空,索引结点才能后移
node = node->next;
}
/// 有一个链表遍历完了,那么把另一个链表的所有结点插入到新的链表中
if(first == NULL)
{
node->next = second;
}
else
{
node->next = first;
}
return result;
}