用 JavaScript 实现链表操做 - 13 Shuffle Merge

TL;DR

把两个链表洗牌合并成一个,系列目录见 前言和目录node

需求

实现函数 shuffleMerge() 把两个链表合并成一个。新链表的节点是交叉从两个链表中取的。这叫洗牌合并。举个例子,当传入的链表为 1 -> 2 -> 3 -> null7 -> 13 -> 1 -> null 时,合并后的链表为 1 -> 7 -> 2 -> 13 -> 3 -> 1 -> null 。若是合并过程当中一个链表的数据先取完了,就从另外一个链表中取剩下的数据。这个函数应该返回一个新链表。git

var first = 3 -> 2 -> 8 -> null
var second = 5 -> 6 -> 1 -> 9 -> 11 -> null
shuffleMerge(first, second) === 3 -> 5 -> 2 -> 6 -> 8 -> 1 -> 9 -> 11 -> null

若是参数之一为空,应该直接返回另外一个链表(即便另外一个链表也为空),不须要抛异常。github

递归解法 1

代码以下:segmentfault

function shuffleMerge(first, second) {
  if (!first || !second) return first || second

  const list = new Node(first.data, new Node(second.data))
  list.next.next = shuffleMerge(first.next, second.next)
  return list
}

解题思路是,首先判断是否有一个链表为空,有就返回另外一个,结束递归。这个判断过了,下面确定是两个链表都不为空的状况。咱们依次从两个链表中取第一个节点组合成新链表,而后递归 shuffleMerge 两个链表的后续节点,并把结果衔接到 list 后面。这段代码基本跟题目描述的意思一致。函数

递归解法 2

在上面的基础上咱们还能作个更聪明的版本,代码以下:测试

function shuffleMergeV2(first, second) {
  if (!first || !second) return first || second
  return new Node(first.data, shuffleMerge(second, first.next))
}

经过简单的调换 firstsecond 的顺序,咱们能把递归过程从 “先取 first 的首节点再取 second 的首节点” 变成 “总老是取 first 的首节点” 。解法 1 中的三行代码简化成了一行。指针

循环解法

循环其实才是本题的考点,由于这题主要是考指针(引用)操做。尤为是把 “依次移动两个链表的指针” 写进一个循环里。不过上个解法中调换两个链表顺序的方式也能够用到这里。代码以下:code

function shuffleMergeV3(first, second) {
  const result = new Node()
  let pr = result
  let [p1, p2] = [first, second]

  while (p1 || p2) {
    if (p1) {
      pr.next = new Node(p1.data)
      pr = pr.next
      p1 = p1.next
    }
    [p1, p2] = [p2, p1]
  }

  return result.next
}

首先咱们生成一个 dummy node result ,同时创建一个 pr 表明 result 的尾节点(方便插入)。两个链表的指针分别叫 p1p2 。在每次循环中咱们都把 p1 的节点数据写到 result 链表的末尾,而后修改指针指向下一个节点。经过 12 行的调换指针,咱们能够保证下一次循环就是对另外一个链表进行操做了。这样一直遍历到两个链表末尾,返回 result.next 结束。递归

参考资料

Codewars Kata
GitHub 的代码实现
GitHub 的测试get

相关文章
相关标签/搜索