实验11-2-5 链表拼接（PTA）

贾值连城2024-03-18 13:40

题目：

本题要求实现一个合并两个有序链表的简单函数。链表结点定义如下：

复制代码

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){
    if(list1==NULL){//如果有一个链表为空，直接返回另一个链表
        return list2;
    }
    if(list2==NULL){
        return list1;
    }
    struct ListNode *p,*head=NULL,*s;//s为新的合并链表
    while(list1!=NULL && list2!=NULL){
        p=(struct ListNode*)malloc(sizeof(struct ListNode));//新建一个结点p
        if(list1->data > list2->data){//p为较小的数
            p->data = list2->data;
            p->next = NULL;
            list2=list2->next;
        }
        else{
            p->data = list1->data;
            p->next = NULL;
            list1=list1->next;
        }
        if(head==NULL){//处理第一个结点
            head=p;
        }
        else{
            s->next=p;//链接上p结点
            
        }
        s=p;
    }
    if(list1==NULL){//当一个链表已经遍历完，直接链接上另一个链表
        s->next=list2;
    }
    else if(list2==NULL){
        s->next=list1;
    }
    return head;
}