题解 | #合并两个排序的链表#

合并两个排序的链表

https://www.nowcoder.com/practice/d8b6b4358f774294a89de2a6ac4d9337

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 * };
 */

/**
 * 
 * @param pHead1 ListNode类 
 * @param pHead2 ListNode类 
 * @return ListNode类
 */
struct ListNode* Merge(struct ListNode* pHead1, struct ListNode* pHead2 ) {
    // write code here   
    int flag;
    struct ListNode *p = pHead1;
    struct ListNode *q = pHead2;
    struct ListNode *temp;

    if(pHead1 == NULL && pHead2 != NULL)
    {
        return pHead2;
    }
    else if(pHead1 != NULL && pHead2 == NULL)
    {
        return pHead1;
    }
    else if(pHead1 == NULL && pHead2 == NULL)
    {
        return NULL;
    }

    struct ListNode *head;
    if(p->val > q->val)
    {
        head = q;
        q = q->next;
        flag = 1;
    }
    else {
        head = p;
        p = p->next;
        flag = 0;
    }

    while(1)
    {
        if(p == NULL){
            head->next = q;
            break;
        }

        if(q == NULL){
            head->next = p;
            break;
        }

        if(p->val > q->val)
        {
            temp = q->next;
            head->next = q;
            q->next = p;
            q = temp;
        }
        else {
            temp = p->next;
            head->next = p;
            p->next = q;
            p = temp;
        }
        head = head->next;        
    }

    if(flag)
    {
        return pHead2;
    }else{
        return pHead1;
    }

}

#合并链表#
全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务