仅使用纯函数反转对象

Question

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

{
  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;
}

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

Answer 1

好吧，它是一种纯数据转换，因此您绝对可以在不使用可变变量的情况下进行。让我们在 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)

很好的小面试问题:)