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

合并两个排序的链表

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

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * comment: 本题的最大亮点和突破点就是构建虚拟头结点,解决不必要的选择判断
     *          而唯一的代价就是需要new一个虚拟结点。这种方案是突破常规的好方法;
     *          接下来就是如何比较的问题,两两比较,并小的放入虚拟头所在的链表即可;
     * @param pHead1 ListNode类 
     * @param pHead2 ListNode类 
     * @return ListNode类
     */
    ListNode* Merge(ListNode* pHead1, ListNode* pHead2) {
      if (pHead1 == nullptr && pHead2 == nullptr) return nullptr;
      else if (pHead1 != nullptr && pHead2 == nullptr) return pHead1;
      else if (pHead1 == nullptr && pHead2 != nullptr) return pHead2;

      auto virtual_head = new ListNode(-1001);
      auto cur = virtual_head;

      while (pHead1 != nullptr && pHead2 != nullptr) {
        if (pHead1->val <= pHead2->val) {
          cur->next = pHead1;
          pHead1 = pHead1->next;
        } else {
          cur->next = pHead2;
          pHead2 = pHead2->next;
        }
        cur = cur->next;
      }

      if (pHead1 == nullptr)
        cur->next = pHead2;
      else
        cur->next = pHead1;

      auto ret = virtual_head->next;
      delete  virtual_head;

      return ret;
    }
};

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务