剑指Offer_面试题17_合并两个有序链表

题目描述

输入两个单调递增的链表，输出两个链表合成后的链表，当然我们需要合成后的链表满足单调不减规则。

分析：这道题有点类似于面试题4_替换空格中的扩展题目：两个有序数组合并。大体思路上是一致的，代码很类似，分别遍历两个链表，将小的结点链接到新链表表尾。后期处理有点区别就是，合并数组在循环结束后，需要用while把剩余的一个数组中的元素一次加入新数组。而本题只需要用if判断哪个链表不空然后把剩下的直接链接到新链表即可，这是链表不同于顺序表的地方，存储方式不同。

/*
struct ListNode {
	int val;
	struct ListNode *next;
	ListNode(int x) :
			val(x), next(NULL) {
	}
};*/
class Solution {
public:
    ListNode* Merge(ListNode* pHead1, ListNode* pHead2)
    {
        if(!pHead1 || !pHead2)
            return !pHead1 ? pHead2 : pHead1;
        ListNode *head = NULL;
        ListNode *p = NULL;
        ListNode *p1 = pHead1;
        ListNode *p2 = pHead2;
        //确定头元素
        if(pHead1->val < pHead2->val)
        {
            head = pHead1;
            p1 = p1->next;
        }
        else
        {
            head = pHead2;
            p2 = p2->next;
        }
        p = head;    //新链表的指针
        
        while(p1 && p2)
        {
            if(p1->val < p2->val)
            {
                p->next = p1;
                p1 = p1->next;
            }
            else{
                p->next = p2;
                p2 = p2->next;
            }
             p = p->next;
        }
        //这里跟合并数组不一样，数组用while
        if(p1 != NULL) 
        {
            p->next = p1;
        }
        else
            p->next = p2;
        return head;
    }
};

用递归思想来解决问题会很容易，因为每加入一个结点，剩下的又相当于执行一次合并操作，注意终止条件。简洁递归：

/*
struct ListNode {
	int val;
	struct ListNode *next;
	ListNode(int x) :
			val(x), next(NULL) {
	}
};*/
class Solution {
public:
    ListNode* Merge(ListNode* pHead1, ListNode* pHead2)
    {
        if(pHead1 == NULL)
            return pHead2;
        else if(pHead2 == NULL)
            return pHead1;
        
        ListNode *mergeHead = NULL;
        if(pHead1->val < pHead2->val)
        {
            mergeHead = pHead1;
            mergeHead->next = Merge(pHead1->next, pHead2);
        }
        else{
            mergeHead = pHead2;
            mergeHead->next = Merge(pHead1, pHead2->next);
        }
        return mergeHead;
    }
};