kotlin スマートなストリーム記述方法

Question

いつもありがとうございます。立て続けに質問すみません。

質問

kotlin(Javaでもいいのですが)で、以下のような処理をスマートにストリームで記述する方法などありますでしょうか。

{8=2.1, 3=0.5, 4=3.0, 11=1.7} // このMapを...
. ↓
3=0.5, 4=3.0, 8=2.1, 11=1.7 // keyでsortし
. ↓
0=0.5, 1=3.0, 5=2.1, 8=1.7 // keyの最小値でoffsetし(この場合全てのkeyから３を引く)
. ↓
0=0.5, 1=3.5, 5=5.6, 8=7.3 // value を累積値にし、
. ↓
[(0, 0.5), (1, 3.5), (5, 5.6), (8, 7.3)] // List<Pair<Int, Double>> で返す

ソースコード

以下のようにダサい感じでは記述ならわかるのですが、スマートな方法などありましたら。。。

    var a = mapOf(8 to 2.1, 3 to 0.5, 4 to 3.0, 11 to 1.7) // 元のMap
    println("元のMap:" + a)
    var b = mutableListOf<Pair<Int, Double>>()  // 最終型のList
    var c = 0.0
    a.keys.sorted().forEach({
        c += a[it]!!
        b.add(it - a.keys.min()!! to c)
    })
    println("最終型のList:" + b)

実行結果

元のMap:{8=2.1, 3=0.5, 4=3.0, 11=1.7}
最終型のList:[(0, 0.5), (1, 3.5), (5, 5.6), (8, 7.3)]

累積値？以外なら

まあまあスッキリ書ける

var a = mapOf(8 to 2.1, 3 to 0.5, 4 to 3.0, 11 to 1.7) // 元のMap
println(a.toList().sortedBy { it.first }.map { Pair(it.first - a.keys.min()!!, it.second) })

[(0, 0.5), (1, 3.0), (5, 2.1), (8, 1.7)]

Comments

[fuzzball]

式が値を返してくれるなら、var sum = 0.0 を追加して、it.second を sum += it.second にすればいけそうなんですけどね。

[mosa]

そうなんですよねー。ことりんは式が値を返してくれないんですよ。

Answer 1

var a = mapOf(8 to 2.1, 3 to 0.5, 4 to 3.0, 11 to 1.7) // 元のMap
var sum = 0.0
println(a.toList().sortedBy { it.first }.map { Pair(it.first - a.keys.min()!!, if(true){sum+=it.second;sum}else{}) })
//=> [(0, 0.5), (1, 3.5), (5, 5.6), (8, 7.3)]

無理矢理ｗ

Comments

[mosa]

あーｗこれはおもしろいですね。
ifが式であることを利用して、ですね。
セミコロンレスJavaのような面白さがありますね。
ありがとうございます。

[mosa]

こっちもBAにしたかったんですが・・・！

Answer 2

「valueを累積値にし」のところがステートフルな変換処理なので、Streamだけで処理するのは無理そうです。
オフセットがデータソース変更に当たるのかが微妙ですが、そこでいったんリスト化して、累積値はfor文などで求めるしかないです。

Comments

[mosa]

あー、なるほど。１行で書けるかどうかは「ステートレス」かどうかで考えれば良いですね。
ありがとうございます。少し追記しました。