剑指OFFER之合并有序链表（九度OJ1519）

题目描述：

输入两个单调递增的链表，输出两个链表合成后的链表，当然我们需要合成后的链表满足单调不减规则。
(hint: 请务必使用链表。)

 

输入：

输入可能包含多个测试样例，输入以EOF结束。
对于每个测试案例，输入的第一行为两个整数n和m(0<=n<=1000, 0<=m<=1000)：n代表将要输入的第一个链表的元素的个数,m代表将要输入的第二个链表的元素的个数。
下面一行包括n个数t(1<=t<=1000000)：代表链表一中的元素。接下来一行包含m个元素，s(1<=t<=1000000)。

 

输出：

对应每个测试案例，
若有结果，输出相应的链表。否则，输出NULL。

 

样例输入：

5 2

1 3 5 7 9

2 4

0 0

 

样例输出：

1 2 3 4 5 7 9

NULL

 

解题思路：

　　首先给定了两个有序的链表，那么可以考虑使用三个指针，p1指向链表1，p2指向链表2,p3指向合并后的链表3.那么依次扫描链表1和2，每次把小的元素放到p3的后面，p3再指向链表3的尾巴。最后要仔细考虑的问题:

　　1 如果两个链表都为空

if(n1 == 0 && n2 == 0)

            printf("NULL\n");

 

　　2 如果其中一个链表为空

    if(head1->next == NULL){

        res->next = head2->next;

        return ;

    }

    if(head2->next == NULL){

        res->next = head1->next;

        return ;

    }

 

　　3 如果一个链表提前扫描完

        Node *p1;

    Node *p2;

    Node *p3 = res;

    while(head1->next != NULL && head2->next != NULL){

        p1 = head1->next;

        p2 = head2->next;

        if(p1->number < p2->number){

            head1->next = p1->next;

            p1->next = p3->next;

            p3->next = p1;

            p3 = p3->next;

        }else{

            head2->next = p2->next;

            p2->next = p3->next;

            p3->next = p2;

            p3 = p3->next;

        }

    }

    if(head1->next == NULL)

        p3->next = head2->next;

    if(head2->next == NULL)

        p3->next = head1->next;    

 

　　解决了这三个问题，就完成了链表的排序。

代码：

#include <stdio.h>

#include <stdlib.h>

typedef struct node{

    int number;

    struct node * next;

}Node;

void mergeList(Node *res,Node *head1,Node *head2);

int main(){

    int n1,n2;

    int i;

    int temp;

    while(scanf("%d %d",&n1,&n2)!=EOF && n1>=0 && n1<=1000 && n2>=0 && n2<=1000){

        Node *head1 = (Node *)malloc(sizeof(Node));

        Node *head2 = (Node *)malloc(sizeof(Node));

        head1->next = NULL;

        head2->next = NULL;

        Node *tail1 = head1;

        Node *tail2 = head2;

        for(i=0;i<n1;i++){

            scanf("%d",&temp);

            Node *p = (Node *)malloc(sizeof(Node));

            p->next = tail1->next;

            tail1->next = p;

            p->number = temp;

            tail1 = tail1->next;

        }

        for(i=0;i<n2;i++){

            scanf("%d",&temp);

            Node *p = (Node *)malloc(sizeof(Node));

            p->next = tail2->next;

            tail2->next = p;

            p->number = temp;

            tail2 = tail2->next;

        }

        Node *res = (Node *)malloc(sizeof(Node));

        mergeList(res,head1,head2);

        if(n1 == 0 && n2 == 0)

            printf("NULL\n");

        else{

            Node *p = res->next;

            printf("%d",p->number);

            p = p->next;

            while(p != NULL){

                printf(" %d",p->number);

                p = p->next;

            }

            printf("\n");

        }

    }

    return 0;

}

void mergeList(Node *res,Node *head1,Node *head2){

    if(head1->next == NULL){

        res->next = head2->next;

        return ;

    }

    if(head2->next == NULL){

        res->next = head1->next;

        return ;

    }

    Node *p1;

    Node *p2;

    Node *p3 = res;

    while(head1->next != NULL && head2->next != NULL){

        p1 = head1->next;

        p2 = head2->next;

        if(p1->number < p2->number){

            head1->next = p1->next;

            p1->next = p3->next;

            p3->next = p1;

            p3 = p3->next;

        }else{

            head2->next = p2->next;

            p2->next = p3->next;

            p3->next = p2;

            p3 = p3->next;

        }

    }

    if(head1->next == NULL)

        p3->next = head2->next;

    if(head2->next == NULL)

        p3->next = head1->next;

    return ;

}

/**************************************************************

    Problem: 1519

    User: xhalo

    Language: C

    Result: Accepted

    Time:250 ms

    Memory:4212 kb

****************************************************************/