PTA 链表拼接

6-4 链表拼接 （20 分)

题目要求

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

struct ListNode {
    int data;
    struct ListNode *next;
};

函数接口定义：

struct ListNode *mergelists(struct ListNode *list1, struct ListNode *list2);

其中list1和list2是用户传入的两个按data升序链接的链表的头指针；函数mergelists将两个链表合并成一个按data升序链接的链表，并返回结果链表的头指针。

裁判测试程序样例：

#include 
#include 

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 *p1, *head=NULL, *p2, *p, *tail=NULL;
  p1=list1;
  p2=list2;
 //每次判断，插入较小值
  while (p1!=NULL||p2!=NULL)
  {
  //分配内存
    p=(struct ListNode *)malloc(sizeof(struct ListNode));
    //两个链表都还有值的情况
    if (p1 != NULL && p2 != NULL){
    if (p1->data <= p2->data)
    {
      p->data = p1->data;
      p1 = p1->next;
      p->next = NULL;
    }
    else
    {
      p->data = p2->data;
      p2 = p2->next;
      p->next = NULL;
    }
    }
    //list2还有值的情况
    else if (p1 == NULL && p2 != NULL)
    {
      p->data = p2->data;
      p2 = p2->next;
      p->next = NULL;
    }
    //list1还有值的情况
    else
    {
      p->data=p1->data;
      p1=p1->next;
      p->next=NULL;
    }
    if (head==NULL)  head=p;
    else  tail->next = p;
    
    tail = p;
  }
  
  return head;
}