输入两个递增排序的链表,合并这两个链表,并使得新链表保持增序。
分析:
- 此题较为简单。可设置两个指针,分别指向两个链表的头结点。比较值大小,指针分别向后移动比较,直至某个列表为末尾。
- 经看书发现可以用递归思路。第一次比较可得到最小值,此值为合并后链表的尾结点。剩余两段链表继续比较,得到一个次小的结点链上去,依次而下。
代码:
// 单链表结点定义
struct ListNode
{
int value;
ListNode * pNext;
};
ListNode * Merge(ListNode * pHead1, ListNode * pHead2)
{
// 哪个先完,另一个则全部链到已有的合并链表后
// 递归返回条件
if(pHead1 == nullptr)
return pHead2;
if(pHead2 == nullptr)
return pHead1;
// 本次递归的尾结点
ListNode * pMergeHead = nullptr;
if(pHead1->value < pHead2->value)
{
pMergeHead = pHead1;
pMergeHead->pNext = Merge(pHead1->pNext, pHead2);
}
else
{
pMergeHead = pHead2;
pMergeHead->pNext = Merge(pHead1, pHead2->pNext);
}
return pMergeHead;
}
若按照循环合并,则有代码:
// 循环
ListNode * Merge2(ListNode * pHead1, ListNode * pHead2)
{
if(pHead1 == nullptr)
return pHead2;
if(pHead2 == nullptr)
return pHead1;
// 0号结点,为方便后面程序统一化
ListNode pMerge0;
pMerge0.pNext = nullptr;
// 指向当前的较小值
ListNode * ptmp = &pMerge0;
// 不断修改指针,按照从小到大来指
while(pHead1 != nullptr && pHead2 != nullptr)
{
if(pHead1->value < pHead2->value)
{
ptmp ->pNext = pHead1;
pHead1 = pHead1 -> pNext;
ptmp = ptmp -> pNext;
}
else
{
ptmp -> pNext = pHead2;
pHead2 = pHead2 -> pNext;
ptmp = ptmp -> pNext;
}
}
// 最后若谁剩余,统一链到合并表中
if(pHead1 == nullptr)
ptmp->pNext = pHead2;
if(pHead2 == nullptr)
ptmp->pNext = pHead1;
return pMerge0.pNext;
}