Clojure：统计嵌套Map中指定嵌套键对的出现次数

嘿，针对你这个嵌套Map里统计[:A :B]键对次数的需求，确实有比clojure.walk/postwalk更直接高效的实现方式，我给你分享几个实用的思路：

方法一：手动递归遍历（高效通用） #

这种方式手动控制遍历逻辑，只关注Map结构，避免不必要的元素遍历，适合处理大型嵌套结构：

(defn count-a-b-pairs [m]
  (if (map? m)
    ;; 先判断当前Map是否符合[:A :B]键对的条件
    (let [current-match (if (and (map? (:A m)) (:B (:A m))) 1 0)]
      ;; 累加当前匹配数 + 所有子Map的匹配数
      (+ current-match
         (apply + (map count-a-b-pairs (vals m)))))
    0))

;; 测试你的示例
(count-a-b-pairs {:A {:B {:A {:B {:A {:B 0}}}}}}) ; => 3

这个方法会递归遍历所有嵌套的Map，只要某个Map中存在:A键，且:A对应的值是一个包含:B键的Map，就计数+1，完美覆盖所有可能的分支场景。

方法二：用tree-seq简化实现（简洁优雅） #

Clojure的tree-seq可以帮我们快速生成嵌套结构的所有节点，配合过滤统计，代码非常简洁：

(defn count-a-b-pairs [m]
  (count 
   (filter #(and (map? (:A %)) (:B (:A %)))
           (tree-seq map? vals m))))

;; 测试示例
(count-a-b-pairs {:A {:B {:A {:B {:A {:B 0}}}}}}) ; => 3

这里tree-seq map? vals m会生成所有嵌套的Map节点（map?判断是否继续遍历，vals取子节点），然后我们过滤出符合[:A :B]键对条件的节点，最后统计数量即可。这种写法非常符合Clojure的函数式风格，可读性拉满。

对比你最初考虑的postwalk实现 #

如果用clojure.walk/postwalk，通常需要借助atom来维护计数器状态，代码会带副作用，而且postwalk会遍历结构中的每一个元素（包括叶子节点比如数字0），相对上面两种方法效率更低：

(defn count-with-postwalk [m]
  (let [counter (atom 0)]
    (clojure.walk/postwalk
     (fn [x]
       (when (and (map? x) (map? (:A x)) (:B (:A x)))
         (swap! counter inc))
       x)
     m)
    @counter))

这种写法不是不行，但对比递归或tree-seq的方案，既不够简洁，也没有必要遍历所有非Map元素，所以并不是最优选择。

内容的提问来源于stack exchange，提问作者Edmond