本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
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 *mergelists(struct ListNode *list1, struct ListNode *list2)
{
struct ListNode *s,*p=list1,*q=list2,*tail=NULL,*head=NULL;//建立新的链表head~tail
if(p==NULL) return q;//如果某一链表为空,直接返回另一链表
if(q==NULL) return p;
while(q!=NULL&&p!=NULL)//有一个链表为空就退出循环
{
//申请新结点
s=(struct ListNode *)malloc(sizeof(struct ListNode));
s->next=NULL;
if(q->data>p->data)//判断哪个值小,把值小的放进s,同时值小的结点所在链表往后移一个
{
s->data=p->data;
p=p->next;
}
else
{
s->data=q->data;
q=q->next;
}
if(head==NULL)//没有头结点,因此第一个结点要特殊处理
head=s;
else
tail->next=s;
tail=s;
}
//防止循环结束还有一个链表没循环完
if(p)
tail->next=p;
if(q)
tail->next=q;
return head;
}