Submission #4026724


Source Code Expand

import java.io.{InputStream, OutputStream, PrintStream}
import java.util.Scanner

import scala.annotation.tailrec

object Main {

  def main(args: Array[String]): Unit = {
    val solver = new Solver(System.in, System.out)
    solver.solve()
  }

  class Solver(in: InputStream, out: OutputStream) {
    lazy val input = new Scanner(in)

    lazy val output = new PrintStream(out)

    def solve(): Unit = {
      val Array(n, k) = input.nextLine().split(" ").map(_.toInt)
      val xs = input.nextLine().split(" ").map(_.toInt)

      output.println(solve(xs.toSet, k))
    }

    def solve(xs: Set[Int], k: Int): Int = {
      var leftPos = 0
      var rightPos = 0
      var foundCount = 0

      if (xs(0)) {
        foundCount += 1
      }

      if (foundCount >= k) {
        return 0
      }

      1.to(Math.pow(10, 8).toInt).foreach { i =>
        if (xs(i)) {
          rightPos = i
          foundCount += 1
        }

        if (foundCount >= k) {
          val min = leftPos.min(rightPos)
          val max = leftPos.max(rightPos)
          println(min, max)
          return  min * 2 + max
        }

        if (xs(i * -1)) {
          leftPos = i
          foundCount += 1
        }

        if (foundCount >= k) {
          val min = leftPos.min(rightPos)
          val max = leftPos.max(rightPos)
          println(min, max)
          return  min * 2 + max
        }
      }

      0
    }

    @tailrec
    final def gcd(m: Int, n: Int): Int = {
      val (x, y) = if (m > n) (m, n) else (n, m)
      if (y != 0) gcd(y, x % y)
      else x
    }

  }

}

Submission Info

Submission Time
Task C - Candles
User lambdasawa
Language Scala (2.11.7)
Score 0
Code Size 1651 Byte
Status WA
Exec Time 2111 ms
Memory 125828 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 300
Status
AC × 1
WA × 3
AC × 1
WA × 3
TLE × 8
Set Name Test Cases
Sample 0_00.txt, 0_01.txt, 0_02.txt, 0_03.txt
All 0_00.txt, 0_01.txt, 0_02.txt, 0_03.txt, 1_00.txt, 1_01.txt, 1_02.txt, 1_03.txt, 1_04.txt, 1_05.txt, 1_06.txt, 1_07.txt
Case Name Status Exec Time Memory
0_00.txt WA 347 ms 25552 KB
0_01.txt WA 341 ms 25276 KB
0_02.txt AC 339 ms 25400 KB
0_03.txt WA 345 ms 25276 KB
1_00.txt TLE 2111 ms 125484 KB
1_01.txt TLE 2107 ms 125348 KB
1_02.txt TLE 2111 ms 124316 KB
1_03.txt TLE 2111 ms 124460 KB
1_04.txt TLE 2111 ms 125184 KB
1_05.txt TLE 2111 ms 125380 KB
1_06.txt TLE 2111 ms 125828 KB
1_07.txt TLE 2111 ms 123232 KB