ScalaのListで比較処理を行いたい

Question

ListやMapなどScalaの様々なデータ構造を利用してアルゴリズムを学習中

• 平均値を出す割り算の際に、Hの含まれる配列の数.length(or size)で割りたいが方法わからない

ロジックで解決できるのか、ライブラリで解決できるのか模索中ですが参考になる情報などいただけるとありがたいです。

object Main extends App{
    // Here your code !
    val list = List(
        (1,"H",0),
        (2,"H",5),
        (3,"H",4),
        (4,"R",2),
        (5,"R",7),
        (6,"H",1),
        (7,"H",8),
        (8,"H",5),
        (9,"R",2),
        (10,"R",5)
        )

    for((a,b,c) <- list){
        // 実現したいこと
        //  0. aの値はインデックスの役割を果たす
        // １．bが"H"の間、cの値を足し算していく
        // ２．bが"R"になったとき、足し算ストップ
        // ３．それまで足したcの値の平均を出し、平均値をavgとする
        // ４．足し算に関わった各配列が"H"の時のcの値が平均値avgより高い場合のみcを表示する
        // ５．次に"R"が"H"に変わったときに1に戻り、処理を再開する
        
        // 例.
        // (1,"H",0),
        // (2,"H",5),
        // (3,"H",4),
        // (4,"R",2),
        // (5,"R",7)
        
        // 上記の場合、cの値は(0+5+4)で9になる
        // 平均は、9/3(足し算にかかわったHの含まれる配列の数の分)=3
        // aが1から3までの間=bが"H"の間にcの値が平均の3より大きい場合にcを表示したい
        // if (c > avg){
        //  println(c)
        // }
        // ここで、平均値3より大きい5と4が表示されると成功

        // また以下のようにRからHになった際には処理を再開したい
        // (4,"R",2),
        // (5,"R",7),
        // (6,"H",1),
        // (7,"H",8),
        // (8,"H",5),
        // (9,"R",2),
        // (10,"R",5)
        // )

        //最終的にはprintln(c)で5, 4, 8, 5が出力できると成功
    }
}

##　試したコード

途中までの処理としてはこのような雰囲気で書いていました

object Main extends App{
    // Here your code !
    val list = List(
        (1,"H",0),
        (2,"H",5),
        (3,"H",4),
        (4,"R",2),
        (5,"R",7),
        (6,"H",1),
        (7,"H",8),
        (8,"H",5),
        (9,"R",2),
        (10,"R",5)
        )
    var num = 0
    
    for((a,b,c) <- list){

        if (b.contains("H")) {
            num = num + c
        }
        if (b.contains("R")) {
            // ここからわからない
            println(num)
            num = 0
        }
        
    }
}

//結果
//9
//0
//14
//0

Answer 1

素直な解答
質問のコードに素直に追加するとこうなります。

var wlist = List[Int]()
for ((a, b, c) <- list) {
  if (b == "H")
    wlist = wlist :+ c
  if (b == "R") {
    if (!(wlist isEmpty)) {
      val avg = (wlist sum) / (wlist length)
      wlist filter (avg < _) foreach (println(_))
      wlist = List()
    }
  }
}

"H"が続く間、wlistという一時的なリストにcの値を保存します。理由は、cの平均を計算してからでなければ、平均より大きな要素を選択できないからです。

• "H"が続く間、cの値をwlistに順に保存する
• "R"が現れたら、wlistが空でなければ、以下の処理を行う

・wlistの平均を計算する。sumが合計、lengthが個数です。整数除算
・wlistから平均より大きなcを選択する。filterを使います
・選択したcを表示する。foreach、printlnを使います
・wlistを空にする

foldLeft
むりやり畳み込みを使うとこうなります。

val (_, ans) = list.foldLeft((List[Int](), List[Int]())) {
  case ((x, y), (a, b, c)) =>
    (
      if (b == "H") x :+ c else List(),
      if (b == "R" && !x.isEmpty) {
        val avg = x.sum / x.length
        y ++ x.filter(avg < _)
      } else y
    )
}
ans.foreach(println(_))

やっていることは上と同じです。wlistに相当するのが、(x,y)のxです。xには"H"が続く間、cの値が保存されます。yは出力用のリストです。"R"が現れる都度、計算した平均より大きなcが順にyに保存されます。畳み込みと同じ処理は再帰を使ってもできます。

もっと良い解答
あるはず。考えましょう。

パターンマッチング(追記)
畳み込みにパターンマッチングを適用して、この回答を終わります。

val (_, ans2) = list.foldLeft((Nil:List[Int], Nil:List[Int])) {
  case ((x, y), (_, "H", c)) => (x :+ c, y)
  case ((Nil, y), _) => (Nil, y)
  case ((x, y), _) => (Nil, {val avg = x.sum / x.length; y ++ x.filter(avg < _)})
}
ans2.foreach(println(_))

Comments

[teratake27]

私の説明は分かりづらかったと思うのですが、完璧に解釈してご丁寧に回答いただき、ありがとうございます。
また、素早くご回答いただいていたのに、すぐに返信ができず申し訳ございませんでした。
かゆいところに手が届く回答をいただけて大変参考になりました。
scala.language.postfixOpsをimportした記法を初めて拝見したので、こんな書き方もあるのかと目から鱗が落ちるようでした。

現在の私ではxebme様よりも良い回答は出せませんが、両パターンとも使いこなせるように精進したいと思います。ありがとうございました。

[xebme]

この回答のままだと恥ずかしいのでパターンマッチングを追記して終わりにします。

[teratake27]

今の私ではじっくり読んでも難しく、黒魔術のようなコードに感じられます。
さすがにここまで短くシンプルに記述できるとは思いもしませんでした。
1つ1つリファレンスを確認しながら、冬休み中にこの書き方の使い方を理解して自分自身でも動作させられるように頑張ってみたいと思います。
ありがとうございました！