【LeetCode】合并两个有序链表 - 递归 - 迭代

小知识，大挑战！本文正在参与“程序员必备小知识”创作活动。

嗨！~ 大家好，我是YK菌 🐷 ，一个微系前端 ✨，喜欢分享自己学到的小知识 🏹，欢迎关注我呀 😘 ~ [微信号: yk2012yk2012，微信公众号：ykyk2012]

今天来合并两个有序链表，这道题我在 [浦发银行笔试] 中遇到过

21. 合并两个有序链表

将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。

链接： leetcode-cn.com/problems/me…

示例

输入： l1 = [1,2,4], l2 = [1,3,4]
输出： [1,1,2,3,4,4]

【解法一】迭代 - 非递归

我们先使用一种简单好理解的解法,就是迭代解法,首先这两个链表都是有序的,所以可以将直接判断 l1 和 l2 哪一个链表的头节点的值更小，将较小值的节点添加到结果里，当一个节点被添加到结果里之后，将对应链表中的节点向后移一位。

/**
 * Definition for singly-linked list.
 * function ListNode(val, next) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.next = (next===undefined ? null : next)
 * }
 */
 
/**
 * @param {ListNode} l1
 * @param {ListNode} l2
 * @return {ListNode}
 */
var mergeTwoLists = function (l1, l2) {
  // 创建一个头节点，值为 -1 
  const prehead = new ListNode(-1);
  // 定义一个前指针，开始的时候指向头节点
  let prev = prehead;
  // 只要两个链表都没有遍历完，就一直遍历
  while (l1 !== null && l2 !== null) {
    if (l1.val <= l2.val) {
      prev.next = l1;
      l1 = l1.next;
    } else {
      prev.next = l2;
      l2 = l2.next;
    }
    // 前指针后移
    prev = prev.next;
  }
  
  // 退出循环一定是因为某一个链表已经遍历完了，只需要把另一个链表的剩余部分都接到prev后面即可
  if (l1 === null) {
    prev.next = l2;
  } else{
    prev.next = l1;
  }
  
  // 直接返回头节点之后的链表
  return prehead.next;
};

【解法二】递归

接下来我们再实现一种递归的解法

首先找到递归终止条件出口，就是一个链表为空，直接返回另一个链表

if(l1 === null){
    return l2
}else if(l2 === null){
    return l1
}

然后进行一般的判断

判断两个链表的头节点哪一个更小，更小的那个头节点保留，然后递归调用剩余的两个链表

if(l1.val < l2.val){
  l1.next = mergeTwoLists(l1.next, l2)
  return l1
}

可以看下面这个动图来理解一下

/**
 * Definition for singly-linked list.
 * function ListNode(val, next) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.next = (next===undefined ? null : next)
 * }
 */
/**
 * @param {ListNode} l1
 * @param {ListNode} l2
 * @return {ListNode}
 */
var mergeTwoLists = function(l1, l2) {
    if(l1 === null){
        return l2
    }else if(l2 === null){
        return l1
    }else if(l1.val < l2.val){
        l1.next = mergeTwoLists(l1.next, l2)
        return l1;
    }else{
        l2.next = mergeTwoLists(l1, l2.next)
        return l2
    }
};