如何在Clojure中计算两个列表的交集?
内置函数方案
Clojure标准库自带了交集计算的工具函数clojure.set/intersection,刚好满足你只对比顶层元素、不递归遍历的需求,只需要把输入列表转为集合后调用即可:
(require '[clojure.set :as set]) ;; 最简实现 (defn list-intersection [l1 l2] (seq (set/intersection (set l1) (set l2))))
调用示例:(list-intersection '(1 2 3 4) '(2 3 4 5))
返回结果:(2 3 4)(注:集合本身无序,若需要保留原列表的元素顺序,可使用下面的实现)
保序的精简自定义实现
如果要求交集结果的元素顺序和第一个输入列表的出现顺序一致,可以用filter结合集合查询实现,不需要引入额外依赖:
(defn list-intersection [l1 l2] (let [l2-set (set l2)] (filter #(contains? l2-set %) l1)))
这个实现时间复杂度为O(n),仅做顶层元素对比,代码完全符合Clojure的函数式风格,远短于你当前的8行实现。
内容的提问来源于stack exchange,提问作者LionelGoulet
相关产品推荐
相关产品推荐

