javaで効率のよい書き方

Question

いつもお世話になっております。

今回は効率化（？）について質問させていただきます。

###やりたいこと
String型の配列に入っている”2016/07/20”という文字列を使って2つのmapを作りたいんです。

例）

String[] list =
{
"2014/12/01",
"2015/01/01",
//~~~~~中略~~~~~~
"2016/05/01",
}

//上記のものを


//下記の形に動的に書き換えたいのですが...
private Map<Integer,String> years;
private Map<String,List<String>> yearMonths;

years.put("0", "2014");
years.put("1", "2015");
years.put("2", "2016");

yearMonths.put("2015", Arrays.asList("12"));
yearMonths.put("2016", Arrays.asList("1","2",/*中略*/"11","12"));
yearMonths.put("2017", Arrays.asList("1","2","3","4","5"));

自分で作ってみたはいいもののコレは実用化できるものなのか？と思い質問させていただきました。
ちなみに今はyearMonthsの中身が連続していますが後々虫食いになる可能性があるので動的に生成させています。

下記が自作で作ったものです。
※mDが上記のlistの代わりのものです。

//private Map<Integer,String> years;		   		...①
//private Map<String,List<String>> yearMonths;		  ...②
		 
//②のList<String>の部分に入れ込む配列を生成
ArrayList<String> monthlist = new ArrayList<String>(); // ...③

 for (int i = 0; i < mD.size(); i++){

 			 if(i == 0){ /**初回*/
 				 
 				 //最小年の①を作成
 				 years.put(i,mD.get(i).getMonth().substring(0, 4));
 				 
 				 //③に月を追加
 				 monthlist.add(mD.get(i).getMonth().substring(5, 7));
 			
 			 }else{ 	 /**二回目以降*/
 				 
 				
 				 //1つ前の配列と現在の配列で年の部分が同値でないかの判定
 				 if(mD.get(i).getMonth().substring(0, 4).equals(years.get(years.size()-1)) )
 				 {
 					/**同値の場合*/
 					
 					//③に月を追加
 					monthlist.add(mD.get(i).getMonth().substring(5, 7));

				 }else{
					 
					/**同値でない場合*/
					 
					 //今まで月を追加してきた③のリストを使い②を作成する
					 yearMonths.put(mD.get(i - 1).getMonth().substring(0, 4), monthlist);
					 //挿入した③のリストはいらないので初期化
 					 monthlist = new ArrayList<String>();

 					 //次年度の①を作成する
					 years.put(years.size(),mD.get(i).getMonth().substring(0, 4));
					 
					 //③に月を追加
					 monthlist.add(mD.get(i).getMonth().substring(5, 7));
				 }

 				 if(i == mD.size()-1){
                      //最後に今まで月を追加してきた③のリストを使い②を作成する
 					 yearMonths.put(mD.get(i - 1).getMonth().substring(0, 4), monthlist);
            //挿入した③のリストはいらないので初期化
 					 monthlist = new ArrayList<String>();
 				 }
 		 }
 }

いかがでしょうか。

ご教授願います。

###追記
効率について追記させていただきます。

なるべく負荷のかからない形
もしくは
実行時間が短い形
の2点を想定しています。

javaのバージョンは８です。

Comments

[coco_bauer]

「効率」の定義は？　コードが短い（文字数が少ない）、実行時間が短い(実行が早い)など、視点が異なる様々な効率がありうるかと思うのですが。

[settyan]

そうですよね...
何の効率を書かずになんて...
効率について追記させていただきました。
ご指摘ありがとうございました。

[yuba]

Javaのバージョンは8ですか？

[settyan]

ご指摘ありがとうございます。
java8です。

Answer 1

ストリームの部分を、他の言語でのタプルを使うイメージで書いてみました。
ただ、SimpleEntryを使うのはたぶんあまり一般的じゃないですね。

yearsの方は、ストリームにするとかなり見苦しくなりそうなので単純なループにしました。

効率が良いかどうかはなんとも言えません。

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

String[] list = { "2014/12/01", "2015/01/01", "2015/02/01", "2015/03/01", "2015/04/01", "2015/05/01",
        "2015/06/01", "2016/05/01", };

Map<Integer, String> years = new HashMap<>();
Map<String, List<String>> yearMonths = Stream.of(list).map(x -> {
    String[] a = x.split("/");
    return new SimpleEntry<String, String>(a[0], a[1]);
}).collect(Collectors.groupingBy(SimpleEntry::getKey,
                                 Collectors.mapping(SimpleEntry::getValue,
                                                    Collectors.toList())));

List<String> yearList = new ArrayList<>(yearMonths.keySet());
Collections.sort(yearList);
for (int i = 0, n = yearList.size(); i < n; i++) {
    years.put(i, yearList.get(i));
}

System.out.println(years);
// => {0=2014, 1=2015, 2=2016}
System.out.println(yearMonths);
// => {2016=[05], 2015=[01, 02, 03, 04, 05, 06], 2014=[12]}

Comments

[settyan]

ご解答ありがとうございました！
試行錯誤してみます！

Answer 2

皆様ご解答ありがとうございました。
様々な書き方を頂いたのですべて試してみようと思います。
僕自身これが言いとまだ決まったわけではないので

ベストアンサーを決めず、自己解決にさせていただきます。
ありがとうございました。

Answer 3

年の順番付きのリストをなぜMapでやる必要があるのかわからなかったため、LinkedHashMapに一任しました。一応年のListへの変換もしておきました。
Mapへの追加は、computeIfAbsentを使えばキーが入ってるかなどの判定がいらずに1行で済みます。

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

public class Q41805 {

	public static void main(String[] args) {
		List<String> years = new ArrayList<>();
		Map<String, List<String>> yearMonths = new LinkedHashMap<>();
		String[] list = {
				"2014/12/01",
				// 中略(連続した年月)
				"2016/05/01",
		};
		for (String date : list) {
			String[] line = date.split("/");
			yearMonths.computeIfAbsent(line[0], k -> new ArrayList<>()).add(line[1]);
		}
		years.addAll(yearMonths.keySet());
		System.out.println(yearMonths);
		System.out.println(years);
	}

}

実行結果

{2014=[12], 2015=[01, 02, 03, 04, 05, 06, 07, 08, 09, 10, 11, 12], 2016=[01, 01, 02, 03, 04, 05]}
[2014, 2015, 2016]

Comments

[settyan]

LinkedHashMapなるものがあるんですね...
ご解答ありがとうございました！
試行錯誤してみます！

Answer 4

手元にJava8の環境がないため、Java7で動作確認してあります。

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.SortedSet;
import java.util.TreeSet;

public class Hoge {

    private String[] list = {
            "2014/12/01",
            "2015/01/01",
            "2014/01/20",
            "2016/01/01",
            "2016/02/01",
            "2016/03/01",
            "2016/04/01",
            "2016/05/01",
            "2016/06/01",
            "2016/07/01",
            "2016/08/01",
            "2016/09/01",
            "2016/10/01",
            "2016/11/01",
            "2016/12/01",
            "2016/12/02",
    };

    private Map<Integer, String> years = new HashMap<>();
    private Map<String, List<String>> yearMonths = new HashMap<>();

    public void doSomething() {

        SortedSet<String> tmpYears = new TreeSet<>();
        Map<String, SortedSet<String>> tmpYearMonths = new HashMap<>();

        for (String line : list) {
            String[] splitted = line.split("\\/");

            tmpYears.add(splitted[0]);

            if (tmpYearMonths.containsKey(splitted[0])) {
                SortedSet<String> months = tmpYearMonths.get(splitted[0]);
                months.add(splitted[1]);
            } else {
                SortedSet<String> months = new TreeSet<>();
                months.add(splitted[1]);
                tmpYearMonths.put(splitted[0], months);
            }
        }

        int i = 0;
        for (String line : tmpYears) {
            years.put(i++, line);
        }

        for (Map.Entry<String, SortedSet<String>> entry : tmpYearMonths.entrySet()) {
            yearMonths.put(entry.getKey(), Arrays.asList(entry.getValue().toArray(new String[] {})));
        }

        System.out.println(years);
        System.out.println(yearMonths);
    }
}

実行結果

{0=2014, 1=2015, 2=2016}
{2014=[01, 12], 2015=[01], 2016=[01, 02, 03, 04, 05, 06, 07, 08, 09, 10, 11, 12]}

Comments

[settyan]

ご解答ありがとうございました！
試行錯誤してみます！

Answer 5

		final Map<String, List<String>> yearMonths = Stream.of(list)
				.map(row -> row.split("/"))
				.collect(Collectors.groupingBy(row -> row[0]))
				.entrySet()
				.stream()
				.collect(Collectors.toMap(
						g -> g.getKey(),
						g -> g.getValue().stream()
								.map(row -> row[1])
								.collect(Collectors.toList())));

Java8だとこういう書き方をするんですが、うーん、これはこれでわかりにくいやつですね⋯

---

追記。Java7以前の機能だけを使った書き方だとこうなります。

		Map<String, List<String>> yearMonths = new HashMap<>();
		for (String line: list) {
			String[] date = line.split("/");
			String year = date[0], month = date[1];
			List<String> monthList = yearMonths.get(year);
			if (monthList == null) {
				monthList = new ArrayList<>();
				yearMonths.put(year, monthList);
			}
			monthList.add(month);
		}

Comments

[settyan]

java8だとそんな書き方があるんですね...
ご解答ありがとうございました！
試行錯誤してみます！

Answer 6

Map<String,List<String>> yearMonths =new Map<>();
for(String date : mD){
   String[] dateArray = date.split("¥/");
   if(!yearMonths.contains(dateArray[0]){
   yearMonths.put(dateArray[0],new ArrayList<String>());
}
yearMonths.get(dateArray[0]).add(dateArray[1]);
}

あとはyearMonthsのkeyを並び替えるとyearsが取得できます。

動作確認して無いですけど、多分こんな感じかな。速さとかはわからないですが。

Comments

[settyan]

ご解答ありがとうございました！
試行錯誤してみます！