単語ごとの出現回数を出現順に表示するには、HashMapとArrayListが必要か

Question

単語ごとの出現回数を出現順に表示するには、HashMapとArryaListが必要でしょうか。
下記の例よりももっとスマートかつコード量が少なく使用するメモリの量が少ない方法があるはずです。

単語ごとの出現回数をカウントするのは、HashMapでやってみた。
・Integer以外にカウントする方法がありますでしょうか。
・出現順に表示するために出現順は、ArrayListでキー管理しましたが、HashMapで追加されたKeyの順序を覚えていればその順序で表示ができるのですが。

下記のコードをもっと改善できる方いますでしょうか。

        String line = "aaa bbb ddd bbb aaa aaa ccc ddd bbb";
        String token[] = line.split(" ");
        HashMap<String,Integer> map = new HashMap<String,Integer>();
        ArrayList list = new ArrayList();
        for(String key: token){
            if(!list.contains(key))list.add(key);
            if(map.get(key) == null ){
                map.put(key,new Integer(1));
            }else{
                map.put(key,new Integer((int)map.get(key)+1));
            }
        }
        for (int i = 0; i < list.size(); i++) {
            StringBuffer a = new StringBuffer();
            a.append(list.get(i));
            a.append(" ");
            a.append(map.get(list.get(i)));
            System.out.println(a);
        }
    }

Answer 1

LinkedHashMapを使えば、取り出す時にIteratorを使って登録した順に取り出すことができます。

不要なお世話かも知れませんが、他の箇所も簡略化できるところを変えてます。

なお、これはJava5以降で使えるサンプルです。

// import java.util.*;

String line = "aaa bbb ddd bbb aaa aaa ccc ddd bbb";
String[] tokens = line.split(" ");
LinkedHashMap<String, Integer> map = new LinkedHashMap<String, Integer>();
for (String token : tokens) {
    if (map.containsKey(token)) {
        map.put(token, map.get(token) + 1);
    } else {
        map.put(token, 1);
    }
}
for (Entry<String, Integer> entry : map.entrySet()) {
    System.out.printf("%s %d%n", entry.getKey(), entry.getValue());
}

Java8でStreamとラムダ式を使うと、さらに簡潔に書けます。

// import java.util.*;
// import java.util.stream.*;

String line = "aaa bbb ddd bbb aaa aaa ccc ddd bbb";
String[] tokens = line.split(" ");
Map<String, Integer> map = Stream.of(tokens)
    .collect(Collectors.toMap(k -> k, v -> 1, (v1, v2) -> v1 + v2, LinkedHashMap::new));
map.entrySet().forEach(x -> System.out.printf("%s %d%n", x.getKey(), x.getValue()));

Answer 2

TreeSetクラスとComparatorインターフェースを使って実装してみました。
JDK1.6で動作確認しています。

import java.util.Comparator;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.TreeSet;

public class WordCount {

    public static void main(String[] args) {

        String line = "aaa bbb ddd bbb aaa aaa ccc ddd bbb";
        String token[] = line.split(" ");

        Map<String, Integer> map = new HashMap<String, Integer>();
        for (String key : token) {
            if (map.containsKey(key)) {
                map.put(key, map.get(key) + 1);
            } else {
                map.put(key, 1);
            }
        }

        // 単語ごとの出現回数を表示
        for (Map.Entry<String, Integer> entry : map.entrySet()) {
            System.out.println(entry.getKey() + " " + entry.getValue());
        }
        
        System.out.println();

        Set<Entry<String, Integer>> set =
                new TreeSet<Entry<String, Integer>>(new CountComparator());
        for (Map.Entry<String, Integer> entry : map.entrySet()) {
            set.add(entry);
        }

        // 出現回数の多い順に並び替えて表示
        for (Map.Entry<String, Integer> entry : set) {
            System.out.println(entry.getKey() + " " + entry.getValue());
        }
    }

    private static class CountComparator implements Comparator<Entry<String, Integer>> {

        @Override
        public int compare(Entry<String, Integer> o1, Entry<String, Integer> o2) {
            if (o2.getValue() != o1.getValue()) {
                return o2.getValue() - o1.getValue();
            }
            return o1.getKey().compareTo(o2.getKey());
        }

    }
}

aaa 3
ddd 2
ccc 1
bbb 3

aaa 3
bbb 3
ddd 2
ccc 1

Comments

[toroleaman]

Comparetor使うのと既存のクラス使うのどっちがいいのでしょうか。
生産性をとるか品質をとるか性能をとるか。
目的によって使い分ける必要がありますね。

Answer 3

まず、質問文にあったコードを短くしてみました。 (↓ の wordcount() メソッド)
その次に LinkedHashMap だけをつかって書いてみました。
出現回数の更新は if ... else と条件分岐せずに、map.getOrDefault(key, 0) + 1) で済ませています。

import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Map.Entry;

public class WordCount {
	public static void main(String[] args) {
		final String line = "aaa bbb ddd bbb aaa aaa ccc ddd bbb";
		wordcount(line);
		System.out.println();
		wordcount2(line);
	}

	//質問文のコードを少し短くした。
	private static void wordcount(String line) {
		final String token[] = line.split(" ");
		HashMap<String, Integer> map = new HashMap<String, Integer>();
		ArrayList<String> list = new ArrayList<String>();

		for (String key : token) {
			if (!list.contains(key)) {
				list.add(key);
			}
			// getOrDefaullt(key, v) は map に key が未登録なら v を返す。
			map.put(key, map.getOrDefault(key, 0) + 1);
		}

		for (String key : list) {
			System.out.printf("%s\t%4d\n", key, map.get(key));
		}
	}

	// LinkedHashMap だけをつかった
	private static void wordcount2(String line) {
		final String token[] = line.split(" ");
		HashMap<String, Integer> map = new LinkedHashMap<String, Integer>();

		for (String key : token) {
			map.put(key, map.getOrDefault(key, 0) + 1);
		}

		for (Entry<String, Integer> entry : map.entrySet()) {
			System.out.printf("%s\t%4d\n", entry.getKey(), entry.getValue());
		}
	}
}

実行例：

aaa	   3
bbb	   3
ddd	   2
ccc	   1

aaa	   3
bbb	   3
ddd	   2
ccc	   1

Comments

[toroleaman]

getOrDefaultは、知りませんでした。ありがとうございます。

Answer 4

配列に入れてソートし，その後単語ごとカウントする方法はどうでしょう？

String line = "aaa bbb ddd bbb aaa aaa ccc ddd bbb";
String token[] = line.split(" ");
Arrays.sort(token);  //java.util.Array
String temp = token[0];
int count = 1;
List<WordCount> list = new ArrayList<WordCount>();
for(int i = 1; i < token.length; i++){
    if(temp.equals(token[i]){
        count++;
    } else {
        list.add(new WordCount(temp, count);
        temp = token[i];
        count = 1;
    }
}
Collections.sort(list);
for(int j = 0; j < list.size(); j++){
    System.out.println(list.get(j));
}

ここで新たにWordCountというクラスを作成して利用しています．

public class WordCount implements Comparable<WordCount>{
    private final String WORD;
    private final int COUNT;
    public WordCount(String word, int count){
        WORD = word;
        COUNT = count;
    }
    public String toString(){
        return WORD + " " + COUNT;
    }
    public int compareTo(WordCount other){
        return other.COUNT - this.COUNT;
    }
}

Comments

[swordone]

「出現順」でしたね，これはちょっと間違ったようです．