Submission #402355

---

Source Code Expand

object Main {
  def main(args: Array[String]): Unit = {
    val sc = new java.util.Scanner(System.in)
    val R,C,K = sc.nextInt
    val N = sc.nextInt
    val r, c = new Array[Int](N)
    // tate(x): x行目に存在する飴の数                                                       
    val tate = new Array[Int](R)
    // yoko(y): y列目に存在する飴の数                                                       
    val yoko = new Array[Int](C)
    // tateAme(x): その行に飴がx個存在する行の数                                            
    val tateAme = new Array[Int](K+1)
    // yokoAme(x): その列に飴がy個存在する列の数                                            
    val yokoAme = new Array[Int](K+1)

    for (i <- 0 to N - 1) {
      r(i) = sc.nextInt - 1
      c(i) = sc.nextInt - 1
      tate(r(i)) += 1
      yoko(c(i)) += 1
    }

    var ans: Int = 0;

    for (i <- 0 to R - 1) {
      if (tate(i) <= K) tateAme(tate(i)) += 1
    }

    for (i <- 0 to C - 1) {
      if (yoko(i) <= K) yokoAme(yoko(i)) += 1
    }

    for (i <- 0 to K) {
      ans += tateAme(i) * yokoAme(K - i)
    }

    for (i <- 0 to N - 1) {
      val sum = tate(r(i)) + yoko(c(i))
      if (sum == K) ans -= 1
      else if (sum == K + 1) ans += 1
    }

    println(ans)
  }
}

Submission Info

Judge Result