本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
struct ListNode {
int data;
struct ListNode *next;
};
函数接口定义:
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2);
其中list1
和list2
是用户传入的两个按data
升序链接的链表的头指针;函数mergelists
将两个链表合并成一个按data
升序链接的链表,并返回结果链表的头指针。
裁判测试程序样例:
#include <stdio.h>
#include <stdlib.h>
struct ListNode {
int data;
struct ListNode *next;
};
struct ListNode *createlist(); /*裁判实现,细节不表*/
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2);
void printlist( struct ListNode *head )
{
struct ListNode *p = head;
while (p) {
printf("%d ", p->data);
p = p->next;
}
printf("\n");
}
int main()
{
struct ListNode *list1, *list2;
list1 = createlist();
list2 = createlist();
list1 = mergelists(list1, list2);
printlist(list1);
return 0;
}
/* 你的代码将被嵌在这里 */
输入样例:
1 3 5 7 -1
2 4 6 -1
输出样例:
1 2 3 4 5 6 7
struct ListNode *head=NULL;
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2){
//函数首先判断两个链表是否都为空,如果都为空,则将head设置为NULL。否则,根据两个链表头节点的数据大小,将head指向较小的链表。
if(list1==NULL&&list2==NULL)
head=NULL;
else{
head=(list1->data<=list2->data)?list1:list2;
//然后定义三个指针p1、p2和p,分别指向两个链表和合并后的链表。
struct ListNode *p1,*p2,*p;
p=head;//指向合并的链表
//指向下一个元素
if(head==list1){
p1=list1->next;
p2=list2;
}
else{
p1=list1;
p2=list2->next;
}
//然后进入一个循环,比较p1和p2指向的节点的数据大小,将较小的节点添加到合并后的链表中,并更新相应的指针。
while(1){
if(p1->data<=p2->data){//注意是哪个指针
p->next=p1;//把目的移动结点连接起来(连接)
p=p1;//存储结点移动到目的移动结点(移动)
p1=p1->next;//移动结点移动到下一个位置(移动)
if(p1==NULL){//当其中一个链表遍历完毕时,将另一个链表剩余的部分添加到合并后的链表末尾。
p->next=p2;
break;
}
}else{
p->next=p2;
p=p2;
p2=p2->next;
if(p2==NULL){//当其中一个链表遍历完毕时,将另一个链表剩余的部分添加到合并后的链表末尾。
p->next=p1;
break;
}
}
}
}//注意括号的范围
return head;
}