仅使用纯函数反转对象

Invert object using only pure functions

本文关键字：对象 函数      更新时间：2023-09-26

假设你有一个这样的对象：

{
  a: [1,2,3],
  b: [2],
  c: [1,4]
}

您需要将其转换为：

{
  1: ['a', 'c'],
  2: ['a', 'b'],
  3: ['a'],
  4: ['c']
}

使用命令式方式和可变数据结构来做到这一点很简单：

function convert (obj) {
  var key, values
    , result = {};
  for (key in obj) {
    values = obj[key];
    values.forEach(function (value) {
      result[value] = result[value] || []; // bad — mutability :(
      result[value].push(key); // bad — mutability :(
    });
  }
  return result;
}

但是有没有办法通过仅使用纯函数而不使用任何赋值来做到这一点？允许使用某些库进行函数式编程。

嗯，这是一个纯数据转换，所以你绝对可以在不使用可变变量的情况下做到这一点。让我们在 Haskell 中做到这一点，它为我们提供了一个静态保证，即没有明确的突变。我们从一个例子开始：

d :: [(Char, [Integer])]
d = [('a', [1,2,3]), ('b', [2]), ('c', [1,4])]

好的，那么新结构的关键是什么？它们是第一个结构的独特元素：

> import Data.List
> let keys = sort . nub . concatMap snd $ d
[1,2,3,4]

然后我们需要知道每个值出现在哪一列中。因此，将每个值与其键配对：

> let pairs = concatMap ('(k,v) -> map (,k) v) d
[(1,'a'),(2,'a'),(3,'a'),(2,'b'),(1,'c'),(4,'c')]

现在我们可以构建结果：

 > [ (k, map snd . filter ((== k) . fst) $ pairs ) | k <- keys ]
 [(1,"ac"),(2,"ab"),(3,"a"),(4,"c")]

在其全部荣耀中：

{-# LANGUAGE TupleSections #-}
import Data.List
rotate d = [ (k, map snd . filter ((== k) . fst) $ pairs ) | k <- keys ]
    where
        keys  = sort . nub . concatMap snd $ d
        pairs = concatMap ('(k,v) -> map (,k) v) d

例如

*A> rotate [('a', [1,2,3]), ('b', [2]), ('c', [1,4])]
[(1,"ac"),(2,"ab"),(3,"a"),(4,"c")]

或者没有命名中间结构的无点版本：

rotate2  = map ('x -> (head (map fst x), map snd x))
         . groupBy ((==) `on` fst)
         . sortBy (comparing fst)
         . concatMap ('(k,v) -> map (,k) v)

很好的小面试问题:)

找到了一种使用 Ramda 库执行此操作的方法：

compose(
  reduce(function(acc, key) {
    return assoc(key, filter(compose(contains(key), propOf(src)), keys(src)), acc);
  }, {}),
  uniq,
  flatten,
  values
)(src)

给定一个像 Haskell Data.Map这样的关联数据结构，我会为每对(v, ks) :: (String, [Int])创建几个只包含 k |-> [v] 的映射（即每个键都映射到包含唯一值的单例列表），然后合并它们。

与Haskell语法保持一致，这会让你得到类似的东西

import qualified Data.Map as Map
insideOut :: (Ord k) => [(v, [k])] -> [(k, [v])]
insideOut vks = Map.toList . Map.unionsWith (++) $
                [ Map.singleton k [v] | (v, ks) <- vks, k <- ks]
相关文章：