Scalaですでにあるリストの中身をマージして新たなリストを作る方法があるでしょうか？

Question

Scala初心者です。
ScalaですでにあるListを処理し、新たなリストを作る方法はあるでしょうか？
やりたいこととしては、SampleClassのListの中身を走査して同じIDだったらマージし、最終的にMargeClassのListに変換するということを目指しています。

public class SampleClass {
    public SampleClass(int id, String str) {
        this.id = id;
        this.str = str;
    }

    public int id;
    public String str;
}

//マージしたものを入れるクラス
public class MargeClass {
    public SampleClass(int id, String str) {
        this.id = id;
        this.str = new ArrayList<>;
        str.add(str);
    }

    public int id;
    public List<String) str;
}

val list = new ArrayList[SampleClass]

//listに値を入れる処理

//同じIDの要素はマージする処理

reduceとかそういうものを使えばいいのかと思いましたが、自分の理解がとぼしくうまく実装できません。ご教授いただけると幸いです。

Answer 1

Scalaの流儀では、更新可能なコンテナーはあまり使いません。

それと、ScalaのListは関数型プログラミングでお馴染みの連結リストです。
ArrayListの代わりなら、Arrayを使った方が良いかもしれません。

また、SampleClassとMargeClassのような情報なら、
新しくクラスを作らなくても、タプルを使って、
Array[(int, String)]からArray[(int, Array[String])]に
変換すれば同じ情報が得られると思います。

途中の状態が分かるように、中間の状態を変数に入れていますが、
メソッドチェインにしてもOKです。

  val a: Array[(Int, String)]
    = Array((1, "aa"), (2, "bb"), (3, "cc"), (2, "dd"), (4, "ee"), (4, "ff"))
  val aa: Map[Int, Array[(Int, String)]] = a groupBy ( _._1 )
//  val aaa = aa.foreach { x => // 訂正: ここの変数は不要でした
  aa.foreach { x =>
    val (id, tuples) = x
    val strings = tuples map ( _._2 ) mkString(", ")
    println(s"id = $id, strings = $strings")
  }

実行結果

id = 2, strings = bb, dd
id = 4, strings = ee, ff
id = 1, strings = aa
id = 3, strings = cc

もっと綺麗に書けるかもしれませんので、研究してみてください。

Comments

[nasum]

回答ありがとうございます。
classを使わないでも実現できるのですね。タプルという機能は恥ずかしながら知らなかったので後で調べてみます。
ありがとうございました。

Answer 2

自分も初心者なので適切な解かどうかはあやしいですが例えばこういうのではいかがでしょうか。

case class SampleClass(val id: Int, val str: String)
case class MergedClass(val id: Int, val str: List[String])

val list = List(SampleClass(1, "a"), SampleClass(2, "b"), SampleClass(1, "c"))
val mergedList = list.groupBy(_.id).map(ent => MergedClass(ent._1, ent._2.map(_.str)))
println(mergedList)

結果出力: List(MergedClass(2,List(b)), MergedClass(1,List(a, c)))

ただし、質問ではArrayListをお使いですがScalaなのでListを使っています。ArrayListにする必要があるならscala.collection.JavaConvertersにあるasJava/asScalaなどでコレクションの変換をしてやるか（暗黙変換をいとわなければ）scala.collection.JavaConversionsの暗黙変換を使って似たような感じにできると思います。

追記:　この方法だとgroupByの結果がMapになるため要素の順番は保存されません。保存する必要があるならまた違った方法になると思いますが、自分だとforeachでしか書けそうもないので多分回答にならないです。

Comments

[nasum]

回答ありがとうございます。
groupByとmapで実現するのですね。要素の順番等には気をつけたいと思います。
ありがとございました。

Answer 3

どうでもいいけど、マージはmergeだけどね...