使用Java流创建嵌套列表中的树图

鉴于: 我有 List<List<Integer>> locations 这是一个位置的坐标。例如,放置A:(2,4),放置B:(5,4),放置C:(10,9),放置D:(2,4)。所以我的 locations 将包含个列表。 我无法更改此格式。

到特定位置的成本是坐标和的平方根。所以要付出的代价是 Place A = Math.sqrt(2 + 4) 要花多少钱 Place B = Math.sqrt(5 + 4) 等等。

产量 :我想得到的是所有地点中成本最低的一个列表。退货要求是 List<List<Integer>> nearestLocations . 我所做的是尝试创建一个 TreeMap<Double, List<List<Integer>>

问题 我的问题是如何用Java 8流转换下面的转换?

 List<List<Integer>> findNearestLocation(int total, List<List<Integer>> allLocations, int size) {
        ArrayList<List<Integer>> results = new ArrayList<>();
        TreeMap<Double, List<Integer>> map = new TreeMap<>();
        for (int i = 0; i < total && i < allLocations.size(); i++) {
            List<Integer> list = allLocations.get(i);
            double l = 0.0;
            for (Integer x : list) {
                l += x * x;
            }
            map.put(Math.sqrt(l), list);
        }
        if (map.size() > 0) {
            for (int get = 0; get < size; get++) {
                results.add(map.get(map.firstKey()));
                map.remove(map.firstKey());
            }

        }
        return results;
    }