結果

問題 No.121 傾向と対策：門松列（その２）
コンテスト
ユーザー gigurururu
提出日時 2015-01-09 22:49:32
言語 Scala(Beta)
(3.8.1)
コンパイル:
scalac _filename_
実行:
java -cp .:/home/linuxbrew/.linuxbrew/Cellar/scala/3.8.1/libexec/maven2/org/scala-lang/scala3-library_3/3.8.1/scala3-library_3-3.8.1.jar:/home/linuxbrew/.linuxbrew/Cellar/scala/3.8.1/libexec/maven2/org/scala-lang/scala-library/3.8.1/scala-library-3.8.1.jar _class_
結果
WA  
実行時間 -
コード長 1,325 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 11,172 ms
コンパイル使用メモリ 293,664 KB
実行使用メモリ 475,868 KB
最終ジャッジ日時 2026-03-09 13:21:15
合計ジャッジ時間 37,086 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge3_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 2 WA * 6 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import scala.io.StdIn

class BIT(size:Int) {
    val n:Int = size
    val buf:Array[Int] = Array.fill(n+1)(0)
    def add(x:Int,v:Int) = {
        var y = x
        while(y<=n){
            buf(y)+=v
            y+=(y& -y)
        }
    }
    def sum(x:Int):Int = {
        var s = 0
        var y = x
        while (y>0){
            s+=buf(y)
            y&=y-1
        }
        return s
    }
}
object Main extends App {
    val n: Int = StdIn.readInt()
    val l: Array[Int] = StdIn.readLine().split(" ").map(_.toInt)
    val bl, br: BIT = new BIT(n)
    //座標圧縮
    val z: Map[Int, Int] = l.sorted.distinct.zipWithIndex.toMap
    val t: Array[Int] = l.map(z(_) + 1)
    //最初は全部右にある
    for (v <- t) br.add(v, 1)

    val cnt: Long = t.zipWithIndex.map { case (v: Int, i: Int) =>
        bl.add(v, 1)
        br.add(v, -1)
        //左右の自分より小さい数同士、大きい数同士を掛けた数がパターン数
        bl.sum(v - 1) * br.sum(v - 1) + (i + 1 - bl.sum(v)) * (n - i - 1 - br.sum(v))
    }.sum
    //左右が同じ数になるパターンを引く 解説(http://ideone.com/qVicVL)の通り
    val exclude: Long = (0 until n).groupBy(t(_)).values.flatMap(a => (1 until a.length).map(i => i * (a.length - i) * (a(i) - a(i - 1) - 1))).sum
    println(cnt - exclude)
}
0