问题描述
本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
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
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
解析
本题要将两个各自有序的链表合并成一个有序的链表,为了图方便直接把新链表建立在原来的链表1中了,即将链表2的各个节点插入到链表1中。依次遍历链表2中的节点,总的来说就是用当前链表2节点的值依次与链表1中节点元素的值比较(通过指针p1的移动来实现),如果p1非空并且p1的元素值小于当前链表2节点的值,p1=p1->next,注意更新p1prior为前一个p1,此处为循环。当跳出循环时,要插入的位置可能为链表1表头、链表1中间或者链表1表尾三种情况,笼统来说,插入位置就是p1前面,具体实现见代码。
代码
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2)
{
struct ListNode *p1=NULL,*p1prior=NULL,*p2=list2,*temp=NULL;
while(p2!=NULL)
{
temp=(struct ListNode *)malloc(sizeof(struct ListNode));
temp->data=p2->data;
temp->next=NULL;
p1=list1;
while(p1!=NULL&&p1->data<temp->data)
{
p1prior=p1;
p1=p1->next;
}
//插到p1前一个位置上
if(p1==list1)
{
temp->next=list1;
list1=temp;
}
else if(p1==NULL) p1prior->next=temp;
else
{
temp->next=p1;
p1prior->next=temp;
}
p2=p2->next;
}
return list1;
}