题目
本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
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 *l,*p,*p1,*p2;
l=(struct ListNode *)malloc(sizeof(struct ListNode));
l->next=NULL;
// p 是用于遍历合并链表的指针,最初指向哨兵节点
p=l;
//PTA中的链表一般没有头结点,第一个结点的值是有效值
p1=list1;
p2=list2;
// 循环比较两个链表当前节点的数据,将较小的节点链接到合并链表中
while(p1!=NULL&&p2!=NULL){
//这里的=解决了运行超时的问题,如果没有的话,遇到相同的数就会死循环
if(p1->data<=p2->data){
p->next=p1; //p 指向 p1,将 p1 指向的节点链接到合并链表
p=p1; //p 移动到新的末尾
p1=p1->next; // p1 前进到下一个节点
}
else if(p2->data<p1->data){
p->next=p2;
p=p2;
p2=p2->next;
}
}
// 如果其中一个链表已经遍历完,直接将另一个链表的剩余部分链接到合并链表的尾部
if(p1!=NULL){
p->next=p1;
}
if(p2!=NULL){
p->next=p2;
}
/ 返回合并后的链表的第一个有效结点
return l->next;
}