本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
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 *p,*p1,*p2,*head1,*head2;
head1=list1;
head2=list2;
while(head2){
//每次循环,针对list2的头结点寻找在list1的插入位置
//该插入位置,为第一个大于它节点data值的位置
//若是插入list1的表头
if(head1!=NULL&&head1->data>head2->data){
p=head2->next;//保存下一节点为新头结点
head2->next=head1;
head1=head2;
head2=p;//恢复list2的头结点,开启下一次循环
} else{
//插入list1的非表头节点,令p1指向待插入位置的前序节点
p1=head1;
p2=head1->next;
while(p2&&p2->data<=head2->data){//找到p2前位插入位置
p1=p2;
p2=p2->next;
}
if(!p2){//p2为空,就插入尾部
p1->next=head2;
head2=head2->next;//插入后list2指向下一节点
p1->next->next=NULL;
}else{
p=head2->next;//临时保存list2的头结点
head2->next=p2;
p1->next=head2;
head2=p;//恢复list2的新头结点
}
}
}
return head1;
}