从相邻元素对还原数组

/*
 * 思路：
 * 1. 使用一个哈希映射来记录每个数的相邻元素。
 * 2. 找到只有一个相邻元素的数，它必定是数组的起点或终点。
 * 3. 使用Java Stream来构建原始数组。
 */
import java.util.*;
import java.util.stream.Collectors;

public class Solution {
    public int[] restoreArray(int[][] adjacentPairs) {
        // 创建一个哈希映射来记录每个数的相邻元素
        Map<Integer, List<Integer>> map = Arrays.stream(adjacentPairs)
                .flatMap(pair -> Arrays.stream(new int[][] { { pair[0], pair[1] }, { pair[1], pair[0] } }))
                .collect(Collectors.groupingBy(pair -> pair[0], Collectors.mapping(pair -> pair[1], Collectors.toList())));

        // 找到只有一个相邻元素的数，作为数组的起点
        int start = map.keySet().stream().filter(key -> map.get(key).size() == 1).findFirst().get();

        // 构建原始数组
        int n = adjacentPairs.length + 1;
        int[] result = new int[n];
        Set<Integer> visited = new HashSet<>();
        result[0] = start;
        visited.add(start);

        for (int i = 1; i < n; i++) {
            int current = result[i - 1];
            int next = map.get(current).stream().filter(neighbor -> !visited.contains(neighbor)).findFirst().get();
            result[i] = next;
            visited.add(next);
        }

        return result;
    }
}