PTA 链表拼接
本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下:
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
我们可以拿出链表二的每个节点然后与链表一的每个节点做比较然后根据题意找到合适的位置插入第一链表,直到把第二链表全部插入第一链表内,然后返回第一链表的头指针就好了。分析如图。
代码实现(c语言)
struct ListNode* mergelists(struct ListNode* list1, struct ListNode* list2)
{
struct ListNode* current1 = list1, * temp1 = NULL;
struct ListNode* current2 = list2, * temp2 = NULL;
struct ListNode* maxcurrent = NULL;//存储最符合条件的链表节点
int flag = 0;
while (current2)//遍历list2
{
maxcurrent = current1;//假设符合条件的节点是list1的链表头
while (current1)//遍历list1
{
if (current2->data > current1->data)
{
//寻找最符合条件的节点并存储
flag = 1;/*标记从list1的链表的中间某个节点符合了条件,
如果没有找到则是0,如果是0,那说明list2的当前节点的数据比list1的每个节点的数据都小,
应该插入list1的头节点前面。*/
maxcurrent = current1;
}
current1 = current1->next;
}
//摘取list2的当前节点
temp2 = current2;
current2 = current2->next;
if (flag)//插入list1的链表中
{
temp1 = maxcurrent->next;
maxcurrent->next = temp2;
temp2->next = temp1;
current1 = maxcurrent;
}
else//插入list1的链表头前面也就是变成list1的新链表的头
{
temp1 = list1;
list1 = temp2;
list1->next = temp1;
current1 = list1;
}
}
return list1;
}