题目:输入两个递增排序的链表,合并这两个链表并使新链表中的节点仍然是按照递增排序的。
链表的定义如下:
struct ListNode{
int value;
ListNode* next;
ListNode(int data=int()):value(data),next(NULL)
{}
};
这是典型的链表归并问题。因为两个链表都是排好序的,所以我们只需要初始化两个指针,分别指向两个链表的第一个节点。并在一个循环中比较两个节点的 value 值,将较小的摘下来放到新链表中去。在循环结束时,肯定会有一个指针走到了其对应链表的尾部,而另一个还没有走到尾部。所以需要将这些节点加入到新链表中去。
代码如下:
ListNode* Merge(ListNode* head1,ListNode* head2)
{
if(head1==NULL)
return head2;
if(head2==NULL)
return head1;
ListNode* newhead=NULL;
ListNode* tail=NULL;
ListNode* tmp=NULL;
while(head1!=NULL && head2!=NULL)
{
if(head1->value > head2->value)
{
tmp=head2;
head2=head2->next;
}
else
{
tmp=head1;
head1=head1->next;
}
if(newhead==NULL)
{
newhead=tmp;
tail=tmp;
}
else
{
tail->next=tmp;
tail=tail->next;
}
}
while(head1!=NULL)
{
if(newhead==NULL)
{
newhead=head1;
tail=head1;
}
else
{
tail->next=head1;
tail=tail->next;
}
head1=head1->next;
}
while(head2!=NULL)
{
if(newhead==NULL)
{
newhead=head2;
tail=head2;
}
else
{
tail->next=head2;
tail=tail->next;
}
head2=head2->next;
}
tail->next=NULL;
return newhead;
}
如果你有任何想法或是可以改进的地方,欢迎和我交流!
完整代码及测试用例在github上:点我前往