我可以用Stream向许多人映射一个元素吗？

Question

是否可以使用Java流将一个集合映射到另一个集合，但第二个集合必须具有与第一个集合不同的元素计数？

我有地图

"1" -> 10, "2-3" -> 20

并希望将其转换为

List<Node> = Node(1,10), Node(2,20), Node(3,20)

我如何使用流来完成这个任务？

我的代码是：

import com.google.common.collect.ImmutableMap;

import java.util.*;

import static java.util.stream.Collectors.toList;

public class MapOneToManyQuestion {

    public static void main(String[] args) {
        new MapOneToManyQuestion().run();
    }

    void run() {
        final Map<String, Integer> map = ImmutableMap.of("1", 10, "2-3", 20);

        List<Node> nodes = map.entrySet().stream().map(entry -> {
            if (Objects.equals(entry.getKey(), "1")) {
                return new Node(1, entry.getValue());
            } else {
                //return new Node(2, entry.getValue());
                //return new Node(3, entry.getValue());
            }
        }).collect(toList());
    }

    class Node {
        private Integer key;
        private Integer value;

        public Node(Integer key, Integer value) {
            this.key = key;
            this.value = value;
        }

        public Integer key() {
            return this.key;
        }

        public Integer value() {
            return this.value;
        }
    }
}

Answer 1

您可以为此使用flatMap。

List<Node> nodes = map.entrySet()
        .stream()
        .flatMap(entry -> {
            String key = entry.getKey();
            Integer value = entry.getValue();
            return Arrays.stream(key.split("-"))
                    .map(splitKey -> new Node(Integer.valueOf(splitKey), value));
        })
        .collect(Collectors.toList());

它通过每个map元素流，并通过-拆分密钥，并为该条目的拆分和映射值为每个键部分创建一个Node对象，最后将其收集到一个列表中。