使用集合和映射来读取文本文件

时间:2017-03-07 22:53:26

标签: java treemaps

作为我的HOMEWORK的一部分,我正在尝试读取由汽车模型组成的文件,并将每个文件的总数输出到控制台。

到目前为止,我已经通过使用带有模型名称作为键并且计数器作为值的Map来实现此目的。

虽然,现在我只想输出添加到Set的模型。 因此,如果菲亚特在文件中,它就不会被使用,因为它不在允许的集合中。

希望这有意义并且有所帮助。

import java.io.File;
import java.io.FileNotFoundException;
import java.util.Map;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeMap;
import java.util.TreeSet;

public class Test 
{

public static void main(String[] args) throws FileNotFoundException 
{
    TreeMap<String, Integer> map = new TreeMap<String, Integer>();
    Set<String> fruit = new TreeSet<String>();
    fruit.add("BMW");
    fruit.add("Mercedes");
    fruit.add("Ford");
    fruit.add("Nissan");
    fruit.add("Tesla");

    File inputFile = new File("input.txt");
    Scanner in = new Scanner(inputFile);

    while (in.hasNext()) 
    {
        String word = in.next();
        if (map.containsKey(word))
        {
            int count = map.get(word) + 1;
            map.put(word,  count);
        }
        else
        {
            map.put(word,  1);
        }
    }
    in.close();

    for (Map.Entry<String, Integer> entry : map.entrySet())
    {
        System.out.println(entry);
    }
  }
}

2 个答案:

答案 0 :(得分:1)

您可以遍历您的设置,然后调用地图&#34;获取&#34;方法

e.g。

for ( String car : fruit ) {
    Integer value = map.get( car );
    if ( value != null ) {
        System.out.println( value );
    }
}

如果结果不为null,则将其打印到控制台,否则,查找下一个。

答案 1 :(得分:0)

试试这个,

TreeMap<String, Integer> map = new TreeMap<String, Integer>();
    Set<String> fruit = new TreeSet<String>();
    fruit.add("BMW");
    fruit.add("Mercedes");
    fruit.add("Ford");
    fruit.add("Nissan");
    fruit.add("Tesla");

    map.put("BMW", 2);
    map.put("Ford", 2);
    map.put("SomeOther", 2);
    System.out.println(map);

    map.keySet().retainAll(fruit);

    System.out.println(map);
    System.out.println(map.keySet());