Scala 任意の値に対して、複数条件でソートできない。

Question

Scalaの基礎を学習中です。
任意のリストに対して、優先順位をつけてソートする実装がわかりません。
（条件：　１の場合はid順でソート、　2の場合は身長順でソート、　3の場合は体重順でソート）

例と、求めたい結果
randomの場合は、idでソートし、idが同じの場合は身長でソート、それでも同じ場合は、体重でソート

// 求めたい結果
  List(Member(1,170,60), Member(2,170,55), Member(3,180,60))

　　　　　　random_2の場合は、身長でソートし、同じの場合はidでソート、それでも同じ場合は体重でソート

// 求めたい結果
  List(Member(3,180,60), Member(1,170,60), Member(2,170,55))

// メンバークラス定義
case class Member(id: Int, height: Int, weight: Int)

// メンバーを生成
val Takahashi = Member(id = 1, height = 170, weight = 60)
val Suzuki = Member(id = 2, height = 170, weight = 55)
val Nakamura = Member(id = 3, height = 180, weight = 60)

// メンバークラスのSeq
val members = Seq(Takahashi, Suzuki, Nakamura)


val random = Seq(1,2,3)
val random_2 = Seq(2,1,3)

def (members: Seq[Member], randomKey: Seq[Int]): Seq[Member] = ???

Answer 1

keyExtractor

ヒントだけですが、何らかのキー抽出が必要ですね。

def keyExtractor(i:Int):Member => Int = i match {
  case 1 => m => m.id
  case 2 => m => -m.height
  case 3 => m => m.weight
  case _ => m => 0
}

sortBy

複数キーのソートはタプルを生成する関数(リテラル)を与える。

members.sortBy (m => (m.id, ... , ...))

sortByだけでなく、sorted,sortWith を利用することも考えてください。

---

追記(Stringも含むソート)

属性がStringも含むソートもという要望に答えて、Ordering[Member]を用意します。結論にまとめました。

---

結論

Ordering[Member]を使う方が汎用性が高いことがわかりました。理由は次のとおり。

• Memeberの項目のみならず、Memberから導出できるどんな項目をも、ソートキーにすることができる
• ソートキーの項目と個数を任意に指定できる

  case class Member(id:Int, name:String, height:Int, weight:Int)

  def flexSort(members:Seq[Member], randomKey:Seq[Int]): Seq[Member] = {
    def comparator(id: Int): Ordering[Member] = id match {
      case 1 => Ordering.by(_.id)
      case 2 => Ordering.by(_.name)
      case 3 => Ordering.by(-_.height)
      case 4 => Ordering.by(_.weight)
      case _ => Ordering.by(_ => 0)
    }
    val actualOrdering = randomKey.map(comparator(_)).reduce((a, o) => Ordering.comparatorToOrdering(a.thenComparing(o)))
    members.sorted(actualOrdering)
  }



  val Yamada = Member(1,"Yamada", 155,60)
  val Yamada2 = Member(2,"Yamada", 170,63)
  val Kawai = Member(3,"Kawai", 181,72)
  val Yamada3 = Member(4,"Yamada", 155,55)
  val list = Seq(Yamada, Yamada2, Kawai,Yamada3)
  val  randomKey = Seq(2,100,-3,3,4)
  val result = flexSort(list,randomKey)
  result.foreach(println(_))

この方法がベストかと問われれば、他にも良い方法があるかもしれない、と思います。

API

scala.math.Ordering
object Ordering
java.util.Comparator

参考

Scala ソート実験
How to sort a list in Scala by two fields?

Comments

[nbki]

回答ありがとうございます。
やってみましたが、実装が難しかったため、再度質問させていただきます。
１、MemberクラスのパラメータにStringがある場合のメソッド
２、keyExtactorの戻り値をsortByにどう使っていいかわからない

[xebme]

1. はヒントの対象外です。条件を変えたければ、質問を変更してください。
Member => Ordering[Member] をちらっと考えましたが、やってみないと分かりません。
2. がこの課題の主題ですから、自分で解くか、解こうとしたコードを質問に追記してください。丸投げでないことを証明しましょう。

Answer 2

そういう文字列を作って、その文字列でソートすればいいのでは？

val members = Seq(Takahashi, Suzuki, Nakamura)

// idでソートし、idが同じの場合は身長でソート、それでも同じ場合は、体重でソート 
scala> members.sortBy{x => "%04d %04d %04d".format(x.id, x.height, x.weight)}
res14: Seq[Member] = List(Member(1,170,60), Member(2,170,55), Member(3,180,60))

// 身長でソートし、同じの場合はidでソート、それでも同じ場合は体重でソート
scala> members.sortBy{x => "%04d %04d %04d".format(x.height, x.id, x.weight)}
res15: Seq[Member] = List(Member(1,170,60), Member(2,170,55), Member(3,180,60))

---

むむ?

random_2の場合は、身長でソートし、同じの場合はidでソート、それでも同じ場合は体重でソート
// 求めたい結果
List(Member(3,180,60), Member(1,170,60), Member(2,170,55))

「身長でソートし」ではなく、「身長を降順ソートし」かよ。

scala>  members.sortBy{x => "%04d %04d %04d".format(10000-x.height, x.id, x.weight)}
res4: Seq[Member] = List(Member(3,180,60), Member(1,170,60), Member(2,170,55))

---

randomの意味役割を理解できていませんでした。そこを理解できたつもりで、追記。

...
val random = Seq(1,2,3)
val random_2 = Seq(2,1,3)

def sss(members: Seq[Member], randomKey: Seq[Int]): Seq[Member] = 
    members.sortBy{
        x => 
        randomKey.map{       
          _ match {
            case 1 => x.id
            case 2 => 10000 - x.height
            case 3 => x.weight
        } 
        }
        .map{"%04d".format(_)}
        .mkString(" ")
    }

val aa = sss(members, random)
println(s" keys: ${random}")
println(s"result: ${aa}")

val bb = sss(members, random_2)
println(s" keys: ${random_2}")
println(s"result: ${bb}")

val random_3 = Seq(3,2,1)

val cc = sss(members, random_3)
println(s" keys: ${random_3}")
println(s"result: ${cc}")


//  keys: List(1, 2, 3)
// result: List(Member(1,170,60), Member(2,170,55), Member(3,180,60))
//  keys: List(2, 1, 3)
// result: List(Member(3,180,60), Member(1,170,60), Member(2,170,55))
//  keys: List(3, 2, 1)
// result: List(Member(2,170,55), Member(3,180,60), Member(1,170,60))

Comments

[nbki]

回答ありがとうございます。
説明が悪いのですが、回答いただいたものは、Seq(1,2,3) , Seq(1,3,2)など考えられる全てのパターンを実装しないといけないため、ソートしたいものが増えて、パターンが増えるほど現実的ではないと思います。
rondom, romdom_2のSeqに対して、数字がどんな順番できても、１つのメソッドで意図した返り値になるようなメソッドにしたいです。

[shiketa]

> 数字がどんな順番できても、１つのメソッドで

追記したものは、どんな順番でもいけてるとおもいますけど。