diff --git a/src/main/java/ru/ilug/c9_most_frequent_words/MainTest.java b/src/main/java/ru/ilug/c9_most_frequent_words/MainTest.java index 8cb6298..bd25632 100644 --- a/src/main/java/ru/ilug/c9_most_frequent_words/MainTest.java +++ b/src/main/java/ru/ilug/c9_most_frequent_words/MainTest.java @@ -8,7 +8,7 @@ import java.util.Map; public class MainTest { public static void main(String[] args) { - MostFrequentWords mostFrequentWords = null; + MostFrequentWords mostFrequentWords = new MostFrequentWordsImpl(); Map result = mostFrequentWords.getNWords("На дворе трава, на траве дрова. Не руби дрова на траве двора!", 3); log.info("3 most frequent words: {}", result); diff --git a/src/main/java/ru/ilug/c9_most_frequent_words/MostFrequentWordsImpl.java b/src/main/java/ru/ilug/c9_most_frequent_words/MostFrequentWordsImpl.java new file mode 100644 index 0000000..b74f65a --- /dev/null +++ b/src/main/java/ru/ilug/c9_most_frequent_words/MostFrequentWordsImpl.java @@ -0,0 +1,24 @@ +package ru.ilug.c9_most_frequent_words; + +import java.util.Arrays; +import java.util.Comparator; +import java.util.List; +import java.util.Map; +import java.util.stream.Collectors; + +public class MostFrequentWordsImpl implements MostFrequentWords { + + @Override + @SuppressWarnings("unchecked") + public Map getNWords(String text, int count) { + return Arrays.stream(text.replaceAll("[.,!]", "").split(" ")) + .collect( + Collectors.groupingBy(c -> c, + Collectors.collectingAndThen(Collectors.toList(), List::size) + ) + ).entrySet().stream() + .sorted(Comparator.comparingInt(e -> ((Map.Entry)e).getValue()).reversed()) + .limit(count) + .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue)); + } +}