通过与一个条目进行比较来查找列表中最接近的条目

时间:2019-04-05 04:17:15

标签: java algorithm sorting data-structures

我有一个adb kill-server adb start-server 类,其中包含许多字段,如下所示:

Unit

我现在有一个public class Unit { private final int id; private final int beds; private final String city; private final double lat; private final double lon; // constructors and getters here // toString method } 列表,它是一个Unit对象,其中包含很多单位。现在,我需要找到从List对象到List的最近单位。限制结果上限。

Unit x

我们在 private List<Unit> nearestUnits(List<Unit> lists, Unit x, int limit) { List<Unit> output = new ArrayList<>(); // how do I sort lists object in such a way so that I can get nearest units here to "x"? return output; } 类中存在经/纬度,因此我们可以用它来计算欧式距离并进行比较。我对如何按最短距离对单位列表进行排序并获取最近的单位感到困惑。到目前为止,我正在使用Java 7,因此无法使用Java 8。

2 个答案:

答案 0 :(得分:2)

//此距离方法参考来自https://stackoverflow.com/a/16794680/6138660

public static double distance(double lat1, double lat2, double lon1,
        double lon2) {
    final int R = 6371; // Radius of the earth

    double latDistance = Math.toRadians(lat2 - lat1);
    double lonDistance = Math.toRadians(lon2 - lon1);
    double a = Math.sin(latDistance / 2) * Math.sin(latDistance / 2)
            + Math.cos(Math.toRadians(lat1)) * Math.cos(Math.toRadians(lat2))
            * Math.sin(lonDistance / 2) * Math.sin(lonDistance / 2);
    double c = 2 * Math.atan2(Math.sqrt(a), Math.sqrt(1 - a));
    double distance = R * c * 1000; // convert to meters

    distance = Math.pow(distance, 2);

    return Math.sqrt(distance);
}

private List<Unit> nearestUnits(List<Unit> lists, Unit x, int limit) {


    lists.sort(new Comparator<Unit>() {

        @Override
        public int compare(Unit o1, Unit o2) {

            double flagLat = x.getLat();
            double flagLon = x.getLon();

            double o1DistanceFromFlag = distance(flagLat, o1.getLat(), flagLon, o1.getLon());
            double o2DistanceFromFlag = distance(flagLat, o2.getLat(), flagLon, o2.getLon());

            return Double.compare(o1DistanceFromFlag, o2DistanceFromFlag);
        }
    });

    return lists.subList(0, limit);;
  }

答案 1 :(得分:2)

您说您知道如何计算距离,因此下面的代码不包含该计算,因此我假设您可以实现calculateDistance()方法。我使用TreeMap来自动对添加到其中的条目进行排序,而类Double实现了Comparable,因此您无需进行排序。 Iterator将返回按计算出的距离排序的键。

private List<Unit> nearestUnits(List<Unit> lists, Unit x, int limit) {
    TreeMap<Double, Unit> sorted = new TreeMap<>();
    List<Unit> output = new ArrayList<>();
    for (Unit unit : lists) {
        Double distance = calculateDistance(unit, x);
        sorted.put(distance, unit);
    }
    Set<Double> keys = sorted.keySet();
    Iterator<Double> iter = keys.iterator();
    int count = 0;
    while (iter.hasNext() && count < limit) {
        Double key = iter.next();
        Unit val = sorted.get(key);
        output.add(val);
        count++;
    }
    return output;
}