本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
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
作者: 张泳
单位: 浙江大学
时间限制: 400 ms
内存限制: 64 MB
代码长度限制: 16 KB
编译器 (1)
C (gcc)
struct ListNode *createlist()
{
struct ListNode *head=NULL, *tail=NULL, *p=NULL;
int data;
while(1){
scanf("%d", &data);
if(data == -1)
return head;
p = (struct ListNode*)malloc(sizeof(struct ListNode));
p->data = data;
p->next = NULL;
if(head){
tail->next = p;
tail = p;
}else{
head = p;
tail = p;
}
}
}
/* 你的代码将被嵌在这里 */
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2)
{
struct ListNode *a = list1, *b = list2;
struct ListNode *c_head = NULL, *c_tail = NULL;
if(!a)
return b;
else if(!b)
return a;
while(a && b){
if(c_head){
if(a->data <= b->data){
c_tail->next =a;
c_tail = a;
a = a->next;
}else{
c_tail->next = b;
c_tail = b;
b = b->next;
}
}else{
if(a->data <= b->data){
c_head = a;
c_tail = a;
a = a->next;
}else{
c_head = b;
c_tail = b;
b = b->next;
}
}
}
c_tail->next = a ? a : b;
return c_head;
}