165. 合并两个排序链表

点击打开链接

合并两个排序链表，使用三个指针，有两个指针分别指需要比较的当前节点，另一个指向要插入的位置的前一个节点。

需要比较两个链表的头节点的值，头节点的值小的可以作为主链表，将另一个链表拆开插入主链表。两种情况的方法是一样的，我们可以写一个通用函数。

class Solution {
public:
    /*
     * @param l1: ListNode l1 is the head of the linked list
     * @param l2: ListNode l2 is the head of the linked list
     * @return: ListNode head of linked list
     */
    ListNode * mergeTwoLists(ListNode * l1, ListNode * l2) {
        // write your code here
    	if(l1 == NULL && l2 != NULL) return l2;
		if(l1 != NULL && l2 == NULL) return l1;
		if(l1 == NULL && l2 == NULL) return NULL;
		ListNode *head;
		if(l1->val < l2->val)
		{
			head = Merge(l1, l2);
		}
		else
		{
			head = Merge(l2, l1);
		}
		return head;
    }
    ListNode *Merge(ListNode *l1, ListNode *l2)
    {
    	if(l1 == NULL && l2 != NULL) return l2;
		if(l1 != NULL && l2 == NULL) return l1;
		if(l1 == NULL && l2 == NULL) return NULL;
		ListNode *p1, *p2;
		ListNode *q;
		p1 = l1;
		p2 = l2;
		q = l1;
    	while(p2 != NULL)
		{
			if(p1->val > p2->val)
			{
				l2 = p2->next;
				p2->next = p1;
				q->next = p2;
				q = q->next;
				p2 = l2;
				continue;
			}
			if(p1->next == NULL && p2 != NULL)
			{
				p1->next = l2;
				break;
			}
			q = p1;
			p1 = p1->next;
		}
		return l1;
	}
};

Merge函数可以表示实现两个链表的合并，再在mergeTwoLists中调用。

经过测试，如果不使用函数调用的方法，在程序运行上耗时会很大，而使用函数Merge的耗时量远小于不使用Merge，所以使用通用函数在程序运行中有很大的好处。

2018/1/25