6-25 链表拼接 (20分)

本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下：

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;
}