合法重新排列数对

/*
 * 思路：
 * 1. 将所有pair转换为Map。
 * 2. 使用Stream从任意一个pair开始查找下一个匹配的pair。
 * 3. 构建最终结果列表。
 */
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class Solution {
    public List<int[]> validReorder(int[][] pairs) {
        // 将pair转换为Map
        Map<Integer, Integer> pairMap = Arrays.stream(pairs)
                .collect(Collectors.toMap(pair -> pair[0], pair -> pair[1]));

        // 使用Stream查找下一个匹配的pair
        List<int[]> result = new ArrayList<>();
        int[] current = pairs[0];
        result.add(current);
        while (pairMap.containsKey(current[1])) {
            current = new int[]{current[1], pairMap.get(current[1])};
            result.add(current);
        }

        return result;
    }
}