本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
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
来源:
来源:PTA | 程序设计类实验辅助教学平台
链接:https://pintia.cn/problem-sets/13/exam/problems/605
提交:
题解:
/*
* 合并两个排序链表
*/
struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2) {
// 创建一个新的头节点作为 list1 与 list2 合并后的头节点
struct ListNode *newHead = (struct ListNode *) malloc(sizeof(struct ListNode));
struct ListNode *cur = newHead;
// 遍历到 l1 或 l2 尾,将节点值较小的节点添加到 cur 后
while (list1 != NULL && list2 != NULL) {
if (list1->data < list2->data) {
cur->next = list1;
list1 = list1->next;
} else {
cur->next = list2;
list2 = list2->next;
}
cur = cur->next;
}
// 将 l1 或 l2 的剩余节点连接到新链表尾
cur->next = list1 == NULL ? list2 : list1;
// newHead 有头节点,不存储数据,返回头节点的下一节点
return newHead->next;
}