本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
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
结尾无空行
我没想到这道题卡住我的居然是两个list相同的情况(测试点2),我想当然的以为需要剔除重复的数字,比如输入两个1 2 3应该输出1 2 3,但实际上,题目不需要剔除重复数字,输入两个1 2 3应当输出1 1 2 2 3 3,这样才能过测试点2。
我的代码:
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2)
{
struct ListNode *temp1=list1;
struct ListNode *temp2=list2;
while(temp2!=NULL){
/*本题目要求相同的list,如两个1 2 3转换成1 1 2 2 3 3的形式
这谁猜的出,正常人都会想着剔除重复数字*/
if(temp1->data<=temp2->data){
if(temp1->next==NULL||temp1->next->data>temp2->data){
//此处申请的空间用来复制temp2的data,方便接下来继续操作temp2
//此处malloc的用法其实不规范,应当检查内存是否申请成功
//我在这里偷懒了一下,下一处malloc函数也偷懒了
struct ListNode *p=(struct ListNode *)malloc(sizeof(struct ListNode));
p->data=temp2->data;
p->next=temp1->next;
temp1->next=p;
temp2=temp2->next;
}else{
temp1=temp1->next;
}
}else if(temp1->data>temp2->data){
/*本题输入的两个list都是升序的,因此temp1->data大于temp2->data
只考虑出现在list1首元节点处*/
struct ListNode *p=(struct ListNode *)malloc(sizeof(struct ListNode));
p->data=temp2->data;
p->next=list1;
list1=p;temp1=p;//更改首元节点list1以及temp1的指向
temp2=temp2->next;
}
}
return list1;
}