Untitled

mail@pastecode.io avatar
unknown
plain_text
a month ago
1.7 kB
0
Indexable
Never
import java.util.*;

public class _09_PopulationCounter {

    public static void main(String[] args) {

        Scanner scanner = new Scanner(System.in);

        Map<String, Map<String, Long>> countries = new LinkedHashMap<>();
        Map<String, Long> countriesTotalCount = new LinkedHashMap<>();

        String input = scanner.nextLine();
        while (!input.equals("report")) {

            String[] tokens = input.split("\\|");
            String city = tokens[0];
            String country = tokens[1];
            long population = Long.parseLong(tokens[2]);

            countries.putIfAbsent(country, new LinkedHashMap<>());
            countriesTotalCount.putIfAbsent(country, 0L);

            countries.get(country).put(city, population);

            long populationToAdd = countriesTotalCount.get(country) + population;
            countriesTotalCount.put(country, populationToAdd);

            input = scanner.nextLine();
        }

        countriesTotalCount.entrySet()
                .stream()
                .sorted(Map.Entry.comparingByValue(Comparator.reverseOrder()))
                .forEach(entry -> {
                    System.out.printf("%s (total population: %d)\n", entry.getKey(), entry.getValue());

                    Map<String, Long> cities = countries.get(entry.getKey());

                    cities.entrySet()
                            .stream()
                            .sorted((entry1, entry2) -> entry2.getValue().compareTo(entry1.getValue()))
                            .forEach(innerMapEntry -> System.out.printf("=>%s: %d\n",
                                    innerMapEntry.getKey(),
                                    innerMapEntry.getValue()));
                });
    }
}
Leave a Comment