题目来源:力扣
题目描述:将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例 1:
输入:l1 = [1,2,4], l2 = [1,3,4] 输出:[1,1,2,3,4,4]
示例 2:
输入:l1 = [], l2 = [] 输出:[]
示例 3:
输入:l1 = [], l2 = [0] 输出:[0]
思路:取两个链表里小的节点插入新的链表,比如示例1的1,2,4和1,3,4,取1插入新节点,取1插入新节点,取2插入新节点,以此类推,直到一个链表全部插完,剩下的那个链表直接链接在新链表的尾部即可。
struct ListNode* mergeTwoLists(struct ListNode* list1, struct ListNode* list2){
if(list1==NULL){
return list2;
}
if(list2==NULL){
return list1;
}
struct ListNode* head=NULL;
struct ListNode* tail=NULL;
while(list1&&list2){
if(list1->val<list2->val){
if(tail==NULL){
head=tail=list1;
}else{
tail->next=list1;
tail=tail->next;
}
list1=list1->next;
}else{
if(tail==NULL){
head=tail=list2;
}else{
tail->next=list2;
tail=tail->next;
}
list2=list2->next;
}
}
if(list1){
tail->next=list1;
}
if(list2){
tail->next=list2;
}
return head;
}
注意:要判断返回空链表。循环控制条件为&&而不是||。插入第一个节点时要判断,因为新链表是空的。
运行结果:
由于代码过于冗杂,我们对其进行优化
优化一:
struct ListNode* mergeTwoLists(struct ListNode* list1, struct ListNode* list2){
if(list1==NULL){
return list2;
}
if(list2==NULL){
return list1;
}
struct ListNode* head=NULL;
struct ListNode* tail=NULL;
if(list1->val<list2->val){
head=tail=list1;
list1=list1->next;
}else{
head=tail=list2;
list2=list2->next;
}
while(list1&&list2){
if(list1->val<list2->val){
tail->next=list1;
list1=list1->next;
}else{
tail->next=list2;
list2=list2->next;
}
tail=tail->next;
}
if(list1){
tail->next=list1;
}
if(list2){
tail->next=list2;
}
return head;
}
我们在外部就解决新链表的头节点问题,同时提取循环内部公共语句tail=tail->next放到循环底部。
优化二:
struct ListNode* mergeTwoLists(struct ListNode* list1, struct ListNode* list2){
if(list1==NULL){
return list2;
}
if(list2==NULL){
return list1;
}
struct ListNode* head=NULL;
struct ListNode* tail=NULL;
head=tail=(struct ListNode*)malloc(sizeof(struct ListNode));
while(list1&&list2){
if(list1->val<list2->val){
tail->next=list1;
list1=list1->next;
}else{
tail->next=list2;
list2=list2->next;
}
tail=tail->next;
}
if(list1){
tail->next=list1;
}
if(list2){
tail->next=list2;
}
struct ListNode* first=head->next;
free(head);
return first;
}
我们给head和tail最开始初始化一个哨兵位的头节点,之后只要进行尾插即可,最后我们释放这个头节点,返回头节点的next
运行结果: