Найти самые близкие записи в списке, сравнивая с одной записью

Question

У меня есть класс 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;
  }

В классе Unit имеется широта / долгота, поэтому мы можем использовать это для вычисления евклидова расстояния и сравнения.Я запутался в том, как отсортировать список единиц по кратчайшему расстоянию и получить ближайшие единицы.Я работаю с Java 7 на данный момент, поэтому я не могу использовать Java 8.

Answer 1

Вы сказали, что знаете, как рассчитать расстояние, поэтому мой код ниже не включает вычисления, поэтому я предполагаю, что вы можете реализовать метод 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;
}

Answer 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);;
  }