Java сортировать карту по количеству значений списка - PullRequest
Новая
       61

Java сортировать карту по количеству значений списка

1 голос
/ 27 января 2020

Например, 

Map<Home, List<People>> ihm = new TreeMap<Home, List<People>>();

и данные выглядят так: 

ihm.put(new Home(...), Arrays.asList(new People(...),
new People(...),
new People(...));
ihm.put(new Home(...), Arrays.asList(new People(...),
new People(...));

Я хочу отсортировать по номеру. людей, живущих в доме.

Как я могу добиться этого с помощью компаратора или сопоставимого?

Ответы [ 3 ]

0 голосов
/ 27 января 2020

Можете ли вы попробовать ниже код? 

public class HomeMain {

public static List<Map.Entry<String, Integer>> sortByValue(Map<String, Integer> wordMap){

    Set<Map.Entry<String, Integer>> set = wordMap.entrySet();
    List<Map.Entry<String, Integer>> list = new ArrayList<Map.Entry<String, Integer>>(set);
    Collections.sort( list, new Comparator<Map.Entry<String, Integer>>()
    {
        public int compare( Map.Entry<String, Integer> o1, Map.Entry<String, Integer> o2 )
        {
            return (o2.getValue()).compareTo( o1.getValue() );
        }
    } );
    return list;
}

public static void main(String[] args) {
    Map<Home, List<People>> ihm = new HashMap<Home, List<People>>();
    ihm.put(new Home("Home"), Arrays.asList(new People(4),
            new People(5),
            new People(6)));

    ihm.put(new Home("Home1"), Arrays.asList(new People(2),
            new People(1),
            new People(9)));

    ihm.put(new Home("Home2"), Arrays.asList(new People(3),
            new People(6),
            new People(2)));

    ihm.put(new Home("Home3"), Arrays.asList(new People(1),
            new People(7),
            new People(6)));
    Map<String, Integer> newMap = new HashMap<String, Integer>();

    Iterator<Map.Entry<Home, List<People>>> itr = ihm.entrySet().iterator();

    while (itr.hasNext()) {
        Map.Entry<Home, List<People>> entry = itr.next();
        List<People> p = entry.getValue();
        int totalPeople = 0;
        for (People people : p) {
            totalPeople += people.getNumberOfPeople();
        }
        newMap.put(entry.getKey().getName(), totalPeople);
    }


    Iterator<Map.Entry<String, Integer>> it1 = newMap.entrySet().iterator();
    System.out.println("UnSorted map:");
    while (it1.hasNext()) {
        Map.Entry<String, Integer> entry = it1.next();
        System.out.println("Key = " + entry.getKey() +
                ", Value = " + entry.getValue());
    }
    List<Map.Entry<String, Integer>> sortedList = sortByValue(newMap);
    System.out.println("Sorted map:");
    for (Map.Entry<String, Integer> entry : sortedList) {
        System.out.println(entry.getKey() + " ====" + entry.getValue());
    }
}} 

public class Home {
  String name;
public Home(String homename){
    this.name=homename;
}}

public class People {
 public int getNumberOfPeople() {
    return numberOfPeople;
}
int numberOfPeople;
    public People(int numOfPeople){
        this.numberOfPeople=numOfPeople;
    }}
0 голосов
/ 27 января 2020

Пример того, как это можно сделать. 

public static void main(String[] args){
    Map<Home, List<People>> ihm = new HashMap<Home, List<People>>();
    ihm.put(new Home(3), Arrays.asList(new People(1), new People(2),new People(3)));
    ihm.put(new Home(2), Arrays.asList(new People(1), new People(4)));
    ihm.put(new Home(4), Arrays.asList(new People(5), new People(4),new People(2),new People(3)));
    ihm.put(new Home(1), Collections.singletonList(new People(5)));

    System.out.println("\nUnSorted Map :");
    for (Map.Entry<Home, List<People>> entry:ihm.entrySet()) {
        System.out.println(entry.getKey());
    }

    Map<Home,List<People>> result = sortByValueCount(ihm);
    System.out.println("\nSorted Map :");
    for (Map.Entry<Home, List<People>> entry:result.entrySet()) {
        System.out.println(entry.getKey());
    }


}

public static Map<Home, List<People>> sortByValueCount(final Map<Home,List<People>> homeListMap) {
    return homeListMap.entrySet()
            .stream()
            .sorted((e1, e2) -> Integer.compare(e2.getValue().size(), e1.getValue().size()))
            .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue, (e1, e2) -> e1, LinkedHashMap::new));
}

static class Home{
    int size;

    public Home(int size) {
        this.size = size;
    }



    @Override
    public String toString() {
        return "Home{" +
                "size=" + size +
                '}';
    }
}

static class People{
    int peopleNumber;

    public People(int peopleNumber) {
        this.peopleNumber = peopleNumber;
    }
}

Для лучшего понимания я установил размер Home равным размеру элементов в нем.

0 голосов
/ 27 января 2020

Это не должно быть сделано как свойство ключа Home, так как вы можете добавлять / удалять людей в Home, портя карту.

Вместо этого сортируйте его динамически: 

ihm.entrySet().stream()
   .sort(Comparator.comparingInt(es -> -es.getValue().size())) // Decreasing; neg. sizes.
   .forEach(es -> System.out.printf("...%n", ...));
...