题目:输入两个递增排序的链表,合并这两个链表并使新链表中的节点仍然是按照递增排序的。
链表的定义如下:
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上:点我前往