合并两个有序的单链表

合并两个有序的单链表

题目描述

给定两个升序的单链表的头节点 head1 和 head2，请合并两个升序链表， 合并后的链表依然升序，并返回合并后链表的头节点。

输入描述：

两个升序的单链表的头节点 head1 和 head2

输出描述：

在给定的函数内返回新链表的头指针。

示例1

输入

5
1 2 3 4 5
6 
7 8 9 10 11 12

输出

1 2 3 4 5 7 8 9 10 11 12

备注：

保证链表的长度不大于1000000

---

题解：

简单的链表合并，但是有一些小技巧：

（1）增加一个哨兵节点可以省很多事；

（2）在一个链表遍历完事后，直接将有序链表尾指针指向另一个链表当前节点，不用再去挨个指一遍了。

代码：

# include <bits/stdc++.h>
using namespace std;

struct list_node{
    int val;
    struct list_node * next;
};


list_node * input_list(void)
{
    int n, val;
    list_node * phead = new list_node();
    list_node * cur_pnode = phead;
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &val);
        if (i == 1) {
            cur_pnode->val = val;
            cur_pnode->next = NULL;
        }
        else {
            list_node * new_pnode = new list_node();
            new_pnode->val = val;
            new_pnode->next = NULL;
            cur_pnode->next = new_pnode;
            cur_pnode = new_pnode;
        }
    }
    return phead;
}


list_node * merge_list(list_node * head1, list_node * head2)
{
    //在下面完成代码
    // trick 1
    list_node *dummy = new list_node();
    dummy->next = NULL;
    
    list_node *p = head1, *q = head2;
    list_node *pre = dummy;
    
    while (p && q) {
        if (p->val <= q->val) {
            pre->next = p;
            p = p->next;
        } else {
            pre->next = q;
            q = q->next;
        }
        pre = pre->next;
    }
    
    // trick 2
    pre->next = p ? p : q;
    
    return dummy->next;
}


void print_list(list_node * head)
{
    while (head != NULL) {
        printf("%d ", head->val);
        head = head->next;
    }
    puts("");
}


int main ()
{
    list_node * head1 = input_list();
    list_node * head2 = input_list();
    list_node * new_head = merge_list(head1, head2);
    print_list(new_head);
    return 0;
}