有没有人知道如何使用内置的collection.sort
和comparator<string>
界面按照频率(从最小到最大)的顺序对单词列表进行排序?
我已经有了一个方法来获取文本文件中某个单词的计数。现在,我只需要创建一个方法来比较每个单词的计数,然后将它们放在按最小频率排序到最大值的列表中。
非常感谢任何想法和提示。我在开始使用这种特殊方法时遇到了麻烦。
public class Parser implements Comparator<String> {
public Map<String, Integer> wordCount;
void parse(String filename) throws IOException {
File file = new File(filename);
Scanner scanner = new Scanner(file);
//mapping of string -> integer (word -> frequency)
Map<String, Integer> wordCount = new HashMap<String, Integer>();
//iterates through each word in the text file
while(scanner.hasNext()) {
String word = scanner.next();
if (scanner.next()==null) {
wordCount.put(word, 1);
}
else {
wordCount.put(word, wordCount.get(word) + 1);;
}
}
scanner.next().replaceAll("[^A-Za-z0-9]"," ");
scanner.next().toLowerCase();
}
public int getCount(String word) {
return wordCount.get(word);
}
public int compare(String w1, String w2) {
return getCount(w1) - getCount(w2);
}
//this method should return a list of words in order of frequency from least to greatest
public List<String> getWordsInOrderOfFrequency() {
List<Integer> wordsByCount = new ArrayList<Integer>(wordCount.values());
//this part is unfinished.. the part i'm having trouble sorting the word frequencies
List<String> result = new ArrayList<String>();
}
}
答案 0 :(得分:7)
首先,您对scanner.next()
的使用似乎不正确。 next()
将返回下一个单词并在每次调用时移到下一个单词,因此以下代码:
if(scanner.next() == null){ ... }
以及
scanner.next().replaceAll("[^A-Za-z0-9]"," ");
scanner.next().toLowerCase();
会消耗,然后只是丢掉一些文字。你可能想做的是:
String word = scanner.next().replaceAll("[^A-Za-z0-9]"," ").toLowerCase();
在while
循环的开头,这样您对单词的更改就会保存在word
变量中,而不会被丢弃。
其次,wordCount
地图的使用略有不同。您要做的是检查地图中是否已有word
来确定要设置的字数。要执行此操作,您应该查看地图,而不是检查scanner.next() == null
,例如:
if(!wordCount.containsKey(word)){
//no count registered for the word yet
wordCount.put(word, 1);
}else{
wordCount.put(word, wordCount.get(word) + 1);
}
或者你可以这样做:
Integer count = wordCount.get(word);
if(count == null){
//no count registered for the word yet
wordCount.put(word, 1);
}else{
wordCount.put(word, count+1);
}
我更喜欢这种方法,因为它更清洁,每个单词只查找一个地图,而第一种方法有时会进行两次查找。
现在,要获得按频率降序排列的单词列表,您可以先将地图转换为列表,然后按照this post中的建议应用Collections.sort()
。以下是适合您需求的简化版本:
static List<String> getWordInDescendingFreqOrder(Map<String, Integer> wordCount) {
// Convert map to list of <String,Integer> entries
List<Map.Entry<String, Integer>> list =
new ArrayList<Map.Entry<String, Integer>>(wordCount.entrySet());
// Sort list by integer values
Collections.sort(list, new Comparator<Map.Entry<String, Integer>>() {
public int compare(Map.Entry<String, Integer> o1, Map.Entry<String, Integer> o2) {
// compare o2 to o1, instead of o1 to o2, to get descending freq. order
return (o2.getValue()).compareTo(o1.getValue());
}
});
// Populate the result into a list
List<String> result = new ArrayList<String>();
for (Map.Entry<String, Integer> entry : list) {
result.add(entry.getKey());
}
return result;
}
希望这有帮助。
修改强> 根据@ dragon66的建议更改了比较功能。感谢。
答案 1 :(得分:1)
您可以从以下内容中比较和提取想法:
public class FrequencyCount {
public static void main(String[] args) {
// read in the words as an array
String s = StdIn.readAll();
// s = s.toLowerCase();
// s = s.replaceAll("[\",!.:;?()']", "");
String[] words = s.split("\\s+");
// sort the words
Merge.sort(words);
// tabulate frequencies of each word
Counter[] zipf = new Counter[words.length];
int M = 0; // number of distinct words
for (int i = 0; i < words.length; i++) {
if (i == 0 || !words[i].equals(words[i-1])) // short-circuiting OR
zipf[M++] = new Counter(words[i], words.length);
zipf[M-1].increment();
}
// sort by frequency and print
Merge.sort(zipf, 0, M); // sorting a subarray
for (int j = M-1; j >= 0; j--) {
StdOut.println(zipf[j]);
}
}
}
答案 2 :(得分:1)
一个解决方案,接近您原来的帖子,其中包含Torious在评论中建议的更正和排序:
import java.util.*;
public class Parser implements Comparator <String> {
public Map<String, Integer> wordCount;
void parse ()
{
Scanner scanner = new Scanner (System.in);
// don't redeclare it here - your attribute wordCount will else be shadowed
wordCount = new HashMap<String, Integer> ();
//iterates through each word in the text file
while (scanner.hasNext ()) {
String word = scanner.next ();
// operate on the word, not on next and next of next word from Scanner
word = word.replaceAll (" [^A-Za-z0-9]", " ");
word = word.toLowerCase ();
// look into your map:
if (! wordCount.containsKey (word))
wordCount.put (word, 1);
else
wordCount.put (word, wordCount.get (word) + 1);;
}
}
public int getCount (String word) {
return wordCount.get (word);
}
public int compare (String w1, String w2) {
return getCount (w1) - getCount (w2);
}
public List<String> getWordsInOrderOfFrequency () {
List<String> justWords = new ArrayList<String> (wordCount.keySet());
Collections.sort (justWords, this);
return justWords;
}
public static void main (String args []) {
Parser p = new Parser ();
p.parse ();
List<String> ls = p.getWordsInOrderOfFrequency ();
for (String s: ls)
System.out.println (s);
}
}
答案 3 :(得分:0)
rodions解决方案是一种泛型地狱,但我没有简单 - 只是不同。
最后,他的解决方案更短更好。
在第一次看来,似乎TreeMap可能是合适的,但它按键进行排序,对按值排序没有帮助,我们无法切换键值,因为我们通过查找键。
所以下一个想法是生成一个HashMap,并使用Collections.sort,但它不需要Map,只需要列表进行排序。在Map中,有一个entrySet,它生成另一个Collection,它是一个Set,而不是List。那是我采取另一个方向的重点:
我实现了一个Iterator:我遍历entrySet,只返回Keys,其值为1.如果值为2,我将它们缓冲以供以后使用。如果Iterator耗尽,我会查看缓冲区,如果它不为空,我将来使用缓冲区的迭代器,增加我寻找的最小值,并创建一个新的Buffer。
Iterator / Iterable对的优点是,可以通过简化的for循环获得这些值。
import java.util.*;
// a short little declaration :)
public class WordFreq implements Iterator <Map.Entry <String, Integer>>, Iterable <Map.Entry <String, Integer>>
{
private Map <String, Integer> counter;
private Iterator <Map.Entry <String, Integer>> it;
private Set <Map.Entry <String, Integer>> buf;
private int maxCount = 1;
public Iterator <Map.Entry <String, Integer>> iterator () {
return this;
}
// The iterator interface expects a "remove ()" - nobody knows why
public void remove ()
{
if (hasNext ())
next ();
}
public boolean hasNext ()
{
return it.hasNext () || ! buf.isEmpty ();
}
public Map.Entry <String, Integer> next ()
{
while (it.hasNext ()) {
Map.Entry <String, Integer> mesi = it.next ();
if (mesi.getValue () == maxCount)
return mesi;
else
buf.add (mesi);
}
if (buf.isEmpty ())
return null;
++maxCount;
it = buf.iterator ();
buf = new HashSet <Map.Entry <String, Integer>> ();
return next ();
}
public WordFreq ()
{
it = fill ();
buf = new HashSet <Map.Entry <String, Integer>> ();
// The "this" here has to be an Iterable to make the foreach work
for (Map.Entry <String, Integer> mesi : this)
{
System.out.println (mesi.getValue () + ":\t" + mesi.getKey ());
}
}
public Iterator <Map.Entry <String, Integer>> fill ()
{
counter = new HashMap <String, Integer> ();
Scanner sc = new Scanner (System.in);
while (sc.hasNext ())
{
push (sc.next ());
}
Set <Map.Entry <String, Integer>> set = counter.entrySet ();
return set.iterator ();
}
public void push (String word)
{
Integer i = counter.get (word);
int n = 1 + ((i != null) ? i : 0);
counter.put (word, n);
}
public static void main (String args[])
{
new WordFreq ();
}
}
由于我的解决方案从stdin读取,因此您使用以下命令调用它:
cat WordFreq.java | java WordFreq