合并两个排序的链表——17

#include <iostream>
#include <assert.h>
using namespace std;
  
template <class T>
struct ListNode//链表结点结构
{
    T _data;
    ListNode<T>* _next;
};
  
template <class T>
ListNode<T>* buy_node(T data)//创建一个链表结点
{
    ListNode<T>* tmp = new ListNode<T>;
    tmp->_data = data;
    tmp->_next = NULL;
    return tmp;
}
  
template <class T>
void init_list(ListNode<T>** node, T data)//初始化链表
{
    *node = buy_node(data);
}
  
template <class T>
void push_node(ListNode<T>*& head, T data)//插入链表结点
{
    if(head == NULL)
    {   
        init_list(&head, data);
        return;
    }   
    ListNode<T>* tmp = head;
    while(tmp->_next != NULL)
    {
        tmp = tmp->_next;
    }
    tmp->_next = buy_node(data);
}
  
template <class T>
void print_list(ListNode<T>* head)//打印链表
{
    while(head != NULL)
    {
        cout<<head->_data<<"->";
        head = head->_next;
    }
    cout<<"NULL"<<endl;
}
  
template <class T>
void destroy_list(ListNode<T>*& head)//销毁链表
{
    if(head != NULL)
    {
        ListNode<T>* cur = head;
        ListNode<T>* tmp = head;
        while(cur != NULL)
        {
            tmp = cur;
            cur = cur->_next;
            delete tmp;
        }
        head = NULL;
    }
}

//实现1：循环
template <class T>
ListNode<T>* MergeList(ListNode<T>* list1, ListNode<T>* list2)
{
    if(list1 == NULL)//条件判断
        return list2;
    else if(list2 == NULL)
        return list1;

    ListNode<T>* newHead = NULL;

    ListNode<T>* tmp1 = NULL;
    ListNode<T>* tmp2 = NULL;

    if(list1->_data <= list2->_data)//决定新链表的头结点
    {
        newHead = list1;
        tmp1 = list1->_next;
        tmp2 = list2;
    }
    else
    {
        newHead = list2;
        tmp1 = list1;
        tmp2 = list2->_next;
    }

    ListNode<T>* cur = newHead;

    while((tmp1 != NULL) && (tmp2 != NULL))
    {
        if(tmp1->_data <= tmp2->_data)//将较小值进行尾插
        {
            cur->_next = tmp1;
            cur = cur->_next;
            tmp1 = tmp1->_next;
        }
        else
        {
            cur->_next = tmp2;
            cur = cur->_next;
            tmp2 = tmp2->_next;
        }
    }

    if(tmp1 == NULL)//将剩下的链表直接接在新链表后面
        cur->_next = tmp2;
    else
        cur->_next = tmp1;

    return newHead;
}

//实现2：递归
template <class T>
ListNode<T>* MergeList(ListNode<T>* list1, ListNode<T>* list2)
{
    if(list1 == NULL)//条件判断
        return list2;
    else if(list2 == NULL)
        return list1;

    ListNode<T>* newHead = NULL;

    if(list1->_data <= list2->_data)
    {
        newHead = list1;
        newHead->_next = MergeList(list1->_next, list2);
    }
    else
    {
        newHead = list2;
        newHead->_next = MergeList(list1, list2->_next);
    }
    return newHead;
}

int main()
{
    ListNode<int>* list1 = NULL;
    push_node(list1, 1);
    push_node(list1, 2);
    push_node(list1, 3);
    push_node(list1, 8);
    push_node(list1, 11);
    push_node(list1, 14);

    ListNode<int>* list2 = NULL;
    push_node(list2, 5);
    push_node(list2, 6);
    push_node(list2, 7);
    push_node(list2, 8);
    push_node(list2, 9);

    cout<<"list1: ";
    print_list(list1);
    cout<<"list2: ";
    print_list(list2);
    
    ListNode<int>* newhead = MergeList(list1, list2);
    cout<<"merge list:";
    print_list(newhead);

    destroy_list(list1);
    destroy_list(list2);

    return 0;
}