Java 同じkeyごとに分けながらHashMapに格納したい。

Question

前提・実現したいこと

お世話になります。

CSVファイルの値を読み込み、HashMapに格納します。
CSVファイルの例は以下の形です。(「,」の左をkeyにします)

1,　100
2,　200
3,　300
1,　200
2,　100
3,　500

格納する時に[1={100, 200}, 2={200, 100}, 3={300, 500}]の形で格納する方法を教えていただきたいです。

発生している問題・エラーメッセージ

get(1)として呼び出した時に 100, 200 と表示させたいのですが、keyが重複した時に値が上書きされてしまい、実行すると　200　と表示されてしまいます。

該当のソースコード

//諸事情で値をStringで格納します。
HashMap<String,String> map = new HashMap<String,String>();

String line;
while((line = br.readLine()) != null) {
 String[] data = line.split(",", 0);
 map.put(data[0],data[1]);
 }

補足情報（FW/ツールのバージョンなど）

Ecilpseを使用しています。

Comments

[TN8001]

外部ライブラリを使用可能であれば、
com.google.common.collect.ArrayListMultimap<K,V>
org.apache.commons.collections4.multimap.ArrayListValuedHashMap<K,V>
のようなズバリなものがあるようです。

Answer 1

無条件にputするのでなく値があるか確認すればよいかと。

	String val = map.get(data[0]);
	if(val == null) {
		map.put(data[0], line);
	} else {
		map.put(data[0], val + ", " + line);
	}

ところで、lineはdata[1]じゃないの？

Answer 2

HashMap<String,List<String>> map = new HashMap<>();

String line;
while((line = br.readLine()) != null) {
 String[] data = line.split(",", 0);
 List<String> list = Optional.ofNullable( map.get(data[0]) ).orElse(new ArrayList<>());
 list.add(data[1]);
 map.put(data[0], _list);
}

Map　の格納変数を配列型にします。

Comments

[momon-ga]

// 諸事情で値をStringで格納します。
とあるのですが、この対応が普通かと思います。

[LouiS0616]

> Optional.ofNullable( map.get(data[0]) ).orElse(new ArrayList<>());

Map#getOrDefaultを使った方が簡潔で分かり良くなるのではないでしょうか。

[t4c62096]

@momon-ga

なるほど、質問者さんの諸事情を理解した回答をすべきかもですね。
参考になります。

だとすれば、@momon-gaさんの回答が正解だと思いました。

[t4c62096]

@LouiS0616
すばらしいです。
確かにそちらのほうが妥当ですね。
ありがとうございます。

[momon-ga]

t4c62096さん
面倒なので、題意に沿う回答しかしませんでしたが
余力があれば、ベターな方法”も”回答するのが良いと思いますよ。

[LouiS0616]

一旦 Map<String, List<String>> を作ってから Map<String, String> に直すのが色々使い勝手が良いのかなと思いました。無茶ぶりされても対処しやすいですし。

[t4c62096]

@LouiS0616
いいね

Answer 3

MapにはcomputeIfAbsentというメソッドがありまして。

HashMap<String,List<String>> map = new HashMap<>();

String line;
while((line = br.readLine()) != null) {
 String[] data = line.split(",", 0);
 map.computeIfAbsent(data[0], k -> new ArrayList<>()).add(data[1]);
 }

Comments

[momon-ga]

HashMap<String,List<String>> map = new HashMap<>(); ですね。

[swordone]

あ、コピーのままにしてました。

[t4c62096]

なるほど。色々きれい、かつ直感的理解につながる書き方を皆さんされていますね
参考になります。