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

合并两个排序的链表

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

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param pHead1 ListNode类 
     * @param pHead2 ListNode类 
     * @return ListNode类
     */
    ListNode* Merge(ListNode* pHead1, ListNode* pHead2) {
        // write code here
        if(pHead1==nullptr) return pHead2;
        else if(pHead2==nullptr) return pHead1;
        ListNode* a, *b, *pre, *res;
        // 保证a指向的是比较小的那个
        if(pHead1->val < pHead2->val){
            a = pHead1;
            b = pHead2;
            res = pHead1;
        }else{
            b = pHead1;
            a = pHead2;
            res = pHead2;
        }
        while(true){
            while(a->val <= b->val && a!=nullptr) {
                pre = a;
                a = a->next; // 指向了第一个大于b的
            }

            if(a==nullptr){ 
                pre->next = b;
                break;
            }else{
                pre->next = b;
                auto tmp = b->next;
                b->next = a;
                pre = b;
                b = tmp;
            }

            if(b == nullptr) break;

        }
        return res;
    }
};







全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务