Как создать карту соседей из заданного списка узлов в Scala?

Question

Предположим, у меня есть List[Node] - список узлов и функция isConnected(n1:Node, n2:Node):Boolean.Теперь я хотел бы создать карту Map[Node, List[Node]], которая отображает каждый узел в список его соседей.

Я пытаюсь выяснить, как использовать groupBy для этого.Имеет ли это смысл?Как бы вы создали карту соседей в идиоматическом Scala?

Answer 1

Как насчет этого?

list.map(node => node -> list.filter(n => isConnected(node, n))).toMap

Answer 2

Предполагая, что l - ваш список:

l zip (l map (n1 => l filter(n2 => isConnected(n1,n2)))) toMap