Java сортирует список с помощью компаратора релевантности
Я пытаюсь отсортировать список музыки по релевантности, соответствующей списку критериев.
public class Music implements Comparable<CriteriaList> {
private String genre, artist, album, titre, price, note;
// getters, setters
public int compareTo(CriteriaList list) {
boolean title, album, genre, artist, note;
title = !list.getTitle().isEmpty() && this.getTitre().equals(list.getTitle());
album = !list.getAlbum().isEmpty() && this.getAlbum().equals(list.getAlbum());
genre = !list.getGenre().isEmpty() && this.getGenre().equals(list.getGenre());
artist = !list.getArtist().isEmpty() && this.getArtist().equals(list.getArtist());
note = !list.getNote().isEmpty() && (Integer.parseInt(this.getNote()) >= Integer.parseInt(list.getNote()));
return ((title ? 1 : 0) + (album ? 1 : 0) + (genre ? 1 : 0) + (artist ? 1 : 0) + (note ? 1 : 0));
}
}
Моя функция compareTo возвращает количество полей, которые соответствуют списку критериев, и проверяет, не являются ли входные данные пустыми.
public class MusicProvider extends Agent {
public List<Music> getMusicsByCL(CriteriaList list) {
ArrayList<Music> res = new ArrayList<Music>();
int[] revelanceTab = new int[res.size()];
int i = 0, revelance;
for (Music music : musicListAvailable) {
revelance = music.compareTo(list);
if (revelance > 1) {
res.add(music);
revelanceTab[++i] = revelance;
}
}
// sort res with revelanceTab
return res;
}
}
Здесь я хочу получить музыку с минимальным значением 1 и отсортировать ее по значению. Как я могу это сделать?
3 ответа
Предполагая, что вы уже создали функциональность для фактического расчета релевантности, я бы поступил так.
Создайте простой класс для хранения музыки и рассчитанного показателя релевантности, просто передайте критерии и сохраните результат расчета.
public class ScoredMusic {
private int relevanceScore;
public ScoredMusic(Music m) { ... }
public void calculateRelevance(Criteria criteria) { ... }
public Music getMusic() { ... }
public int getRelevanceScore() { ... }
}
Затем я оценил бы все имеющиеся у вас экземпляры музыки, сохранил бы их в списке и сделал бы очень простой compareTo()
реализация, которая просто сравнивает relevanceScore
между каждым ScoredMusic
пример.
Comparable используется для сравнения двух экземпляров музыки друг с другом. Если вы хотите сравнить с внешней сущностью, используйте реализацию Comparator и передайте ее в Collections.sort (List, Comparator). Компаратор должен быть инициализирован с помощью CriteriaList, а метод сравнения вернет положительное число, если первый элемент должен быть ранжирован выше, отрицательное число, если второй элемент должен быть ранжирован выше, или 0, если они эквивалентны. В вашем примере вы использовали бы метод CompareTo и вычли результат второго элемента из первого и вернули его.
Что-то вроде этого:
import java.util.Comparator;
public class MusicComparator implements Comparator<Music> {
private final CriteriaList criteria;
public MusicComparator(CriteriaList criteria) {
this.criteria = criteria;
}
@Override
public int compare(Music o1, Music o2) {
return score(o1) - score(o2);
}
private int score(Music music) {
boolean title, album, genre, artist, note;
title = criteria.getTitle().isEmpty() || criteria.getTitle().equals(music.getTitle());
album = criteria.getAlbum().isEmpty() || criteria.getAlbum().equals(music.getAlbum());
genre = criteria.getGenre().isEmpty() || criteria.getGenre().equals(music.getGenre());
artist = criteria.getArtist().isEmpty() || criteria.getArtist().equals(music.getArtist());
note = criteria.getNote().isEmpty() || (!music.getNote().isEmpty() && Integer.parseInt(music.getNote()) >= Integer.parseInt(criteria.getNote()));
return ((title ? 1 : 0) + (album ? 1 : 0) + (genre ? 1 : 0) + (artist ? 1 : 0) + (note ? 1 : 0));
}
}
Кстати, методы isEmpty() не защитят вас от исключений нулевого указателя. Вам понадобится лучший метод обработки, если поля могут быть пустыми.
Это мой последний класс,
public class ScoredMusic implements Comparable<ScoredMusic> {
private int revelanceScore = 0;
private Music music;
public ScoredMusic(Music music, CriteriaList crit) {
this.music = music;
calculateRevelance(crit);
}
private void calculateRevelance(CriteriaList list) {
boolean title, album, genre, artist, note;
title = !list.getTitle().isEmpty() && music.getTitre().equals(list.getTitle());
album = !list.getAlbum().isEmpty() && music.getAlbum().equals(list.getAlbum());
genre = !list.getGenre().isEmpty() && music.getGenre().equals(list.getGenre());
artist = !list.getArtist().isEmpty() && music.getArtist().equals(list.getArtist());
note = !list.getNote().isEmpty() && (Integer.parseInt(music.getNote()) >= Integer.parseInt(list.getNote()));
revelanceScore = ((title ? 1 : 0) + (album ? 1 : 0) + (genre ? 1 : 0) + (artist ? 1 : 0) + (note ? 1 : 0));
}
public Music getMusic() {
return music;
}
public int getRevelanceScore() {
return revelanceScore;
}
public int compareTo(ScoredMusic other) {
return Integer.compare(this.getRevelanceScore(), other.getRevelanceScore());
}
}
и в моем втором классе
public List<ScoredMusic> getMusicsScoredByCL(CriteriaList list) {
ArrayList<ScoredMusic> scoredMusics = new ArrayList<ScoredMusic>();
ScoredMusic sc;
for (Music music : musicListAvailable) {
sc = new ScoredMusic(music, list);
scoredMusics.add(sc);
}
// sort by revelance and descending order
Collections.sort(scoredMusics, Collections.reverseOrder());
return scoredMusics;
}