用 JavaScript 实现链表操作 - 02 Length & Count

链表长度与元素计数

最新推荐文章于 2025-02-16 15:10:49 发布

转载最新推荐文章于 2025-02-16 15:10:49 发布 · 334 阅读

CC 4.0 BY-SA版权

原文链接：https://segmentfault.com/a/1190000007689904

文章标签：

本文介绍如何通过递归和循环方式实现链表长度计算及特定元素出现次数的统计。包括不同版本的length()与count()函数实现。

TL;DR

计算链表的长度和指定元素的重复次数。系列目录见前言和目录。

实现一个 length() 函数来计算链表的长度。

length(null) === 0
length(1 -> 2 -> 3 -> null) === 3

实现一个 count() 函数来计算指定数字在链表中的重复次数。

count(null, 1) === 0
count(1 -> 2 -> 3 -> null, 1) === 1
count(1 -> 1 -> 1 -> 2 -> 2 -> 2 -> 2 -> 3 -> 3 -> null, 2) === 4

递归是最有表达力的版本。思路非常简单。每个链表的长度 length(head) 都等于 1 + length(head.next) 。空链表长度为 0 。

function length(head) {
  return head ? 1 + length(head.next) : 0
}

链表循环第一反应是用 while (node) { node = node.next } 来做，循环外维护一个变量，每次自增 1 即可。

function lengthV2(head) {
  let len = 0
  let node = head

  while (node) {
    len++
    node = node.next
  }

  return len
}

for 和 while 在任何情况下都是可以互换的。我们可以用 for 循环把变量初始化，节点后移的操作都放到一起，简化一下代码量。注意因为 len 要在 for 外部作为返回值使用，我们只能用 var 而不是 let/const 声明变量。

function lengthV3(head) {
  for (var len = 0, node = head; node; node = node.next) len++
  return len
}

跟 length 思路类似，区别只是递归时判断一下节点数据。

function count(head, data) {
  if (!head) return 0
  return (head.data === data ? 1 : 0) + count(head.next, data)
}

这里我直接演示的 for 版本，思路类似就不多说了。

function countV2(head, data) {
  for (var n = 0, node = head; node; node = node.next) {
    if (node.data === data) n++
  }
  return n
}