面试题17:合并两个排序的链表

题目:输入两个递增排序的链表,合并这两个链表并使新链表中的节点仍然是按照递增排序的。

链表的定义如下:

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上:点我前往