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

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

链表的定义如下:

1
2
3
4
5
6
7
struct ListNode{
int value;
ListNode* next;
ListNode(int data=int()):value(data),next(NULL)
{}
};


分析

这是典型的链表归并问题。因为两个链表都是排好序的,所以我们只需要初始化两个指针,分别指向两个链表的第一个节点。并在一个循环中比较两个节点的value值,将较小的摘下来放到新链表中去。在循环结束时,肯定会有一个指针走到了其对应链表的尾部,而另一个还没有走到尾部。所以需要将这些节点加入到新链表中去。

代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
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上:点我前往