Using HashMap(String, ArrayList<String>)

Question

I am trying to make the program read four .txt files and count the words in those files.

private HashMap<String, ArrayList<String>> mapWords = new HashMap<String, ArrayList<String>>();

public void countWordsInFiles() {
    buildWordFileMap();
}

private void buildWordFileMap() {
    mapWords.clear();
    String[] files = {"brief1.txt","brief2.txt","brief3.txt","brief4.txt"};
    
    for(int i=0; i<files.length; i++){
        FileResource resource = new FileResource("data/" + files[i]);
        addWordsFromFile(resource); 
    }
}

private void addWordsFromFile(FileResource resource) {
    for(String word : resource.words()){
        word = word.toLowerCase();
        
        if (mapWords.keySet().contains(word)){
            mapWords.put(word, //Not sure how to use arraylist here);
        }
        else {
            mapWords.put(word, //Not sure how to use arraylist here);
        }
    }
}

The problem is I'm not sure how to make the "if" in the method "addWordsFromFile". Basically, I want my output to be something like this:

The greatest number of files a word appears in is three, and there are two such words: “cats” and “and”.

“cats” appears in the files: brief1.txt, brief3.txt, brief4.txt

“and” appears in the files: brief1.txt, brief3.txt, brief4.txt

You can use mapWords.containsKey instead of mapWords.keySet().contains — OneCricketeer
– OneCricketeer, Commented Nov 24, 2015 at 16:54

Jörn Buitink · Accepted Answer · 2015-11-24 16:57:59Z

3

put a new ArrayList if the key was not found. Afterwards use the arraylist:

private void addWordsFromFile(FileResource resource, String fileName) {
    for(String word : resource.words()){
        word = word.toLowerCase();

        //Make sure key exists and ArrayList is initialized:
        if (!mapWords.containsKey(word)){
            mapWords.put(word, new ArrayList<String>());
        }

        //Add filename to word's ArrayList if filename wasn't added before:
        if (!mapWords.get(word).contains(fileName)) {
            mapWords.get(word).add(fileName);
        }
    }
}

edited Nov 24, 2015 at 16:57

answered Nov 24, 2015 at 16:50

Jörn Buitink

2,9262 gold badges24 silver badges33 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Amir Afghani · Accepted Answer · 2015-11-24 17:03:13Z

0

Consider using a Guava MultiMap so that you don't have to worry about checking for the null ArrayList and maintenance. Now you only have two lines of code to write

ListMultimap<String, String> mapWords = ArrayListMultimap.create();
//assuming resource has a way to get the file name    
mapWords.put(word, resource.getFileName());

answered Nov 24, 2015 at 17:03

Amir Afghani

38.7k17 gold badges87 silver badges127 bronze badges

Collectives™ on Stack Overflow

Using HashMap(String, ArrayList<String>)

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related