Submission #2827473


Source Code Expand

import java.awt.geom.Point2D
import java.io._
import java.util.StringTokenizer

import scala.collection.mutable
import scala.util.Sorting
import math.{abs, max, min}
import scala.collection.mutable.{ArrayBuffer, ListBuffer}
import scala.reflect.ClassTag

object Main {
  val MOD = 1000000007
  val out = new PrintWriter(System.out)

  case class Barrier(x: Int, y: Int, r: Int)
  type WUGraph = Array[Barrier]
  case class Coordinate(x: Int, y: Int)

  def solve(): Unit = {
    val sc = new InputReader(System.in)
    val N = sc.nextInt()
    val A = map(N)(_ => sc.nextLong())
    val ans = map(2) { par =>
      var cnt = 0L
      var sum = 0L
      rep(N) { i =>
        sum += A(i)
        // + でないといけないとき
        if (i % 2 == par && sum <= 0) {
          val ops = 1 - sum
          cnt += ops
          sum += ops
        // -でないといけないとき
        } else if (i % 2 != par && sum >= 0){
          val ops = 1 + sum
          cnt += ops
          sum -= ops
        }
      }
      cnt
    }.min

    out.println(ans)
  }


  def main(args: Array[String]): Unit = {
    solve()
    out.flush()
  }


  class InputReader(val stream: InputStream) {
    private val reader = new BufferedReader(new InputStreamReader(stream), 32768)
    private var tokenizer: StringTokenizer = null

    def next(): String = {
      while (tokenizer == null || !tokenizer.hasMoreTokens)
        tokenizer = new StringTokenizer(reader.readLine)
      tokenizer.nextToken
    }

    def nextInt(): Int = next().toInt
    def nextLong(): Long = next().toLong
    def nextChar(): Char = next().charAt(0)
  }
  def rep(n: Int)(f: Int => Unit): Unit = {
    var i = 0
    while(i < n) { f(i); i += 1 }
  }
  def rep_r(n: Int)(f: Int => Unit): Unit = {
    var i = n - 1
    while(i >= 0) { f(i); i -= 1 }
  }

  def map[@specialized A: ClassTag](n: Int)(f: Int => A): Array[A] = {
    val res = Array.ofDim[A](n)
    rep(n)(i => res(i) = f(i))
    res
  }

  implicit class ArrayOpts[A](val as: Array[A]) extends AnyVal {
    // todo Orderingだとboxing発生するので自作Orderを用意したい
    def maxByOpt[B: Ordering](f: A => B): Option[A] = {
      if (as.nonEmpty) Some(as.maxBy(f)) else None
    }

    def grpBy[K](f: A => K): mutable.Map[K, ArrayBuffer[A]] = {
      val map = mutable.Map.empty[K, ArrayBuffer[A]]
      rep(as.length)(i => map.getOrElseUpdate(f(as(i)), ArrayBuffer()) += as(i))
      map
    }

    def sumBy[B](f: A => B)(implicit num: Numeric[B]): B = {
      var sum = num.zero
      rep(as.length)(i => sum = num.plus(sum, f(as(i))))
      sum
    }

    def minByEx[B](f: A => B, ixRange: Range = as.indices)(implicit cmp: Ordering[B]): (A, B) = {
      limit(f, ixRange)(cmp.lt)
    }

    def maxByEx[B](f: A => B, ixRange: Range = as.indices)(implicit cmp: Ordering[B]): (A, B) = {
      limit(f, ixRange)(cmp.gt)
    }

    private def limit[B](f: A => B, ixRange: Range)(cmp: (B, B) => Boolean): (A, B) = {
      var limA = as(ixRange.head)
      var limB = f(limA)

      for (i <- ixRange.tail) {
        val a = as(i)
        val b = f(a)
        if (cmp(b, limB)) {
          limA = a
          limB = b
        }
      }
      (limA, limB)
    }
  }

  implicit class IterableOpts[A](val as: Iterable[A]) extends AnyVal {
    def sumBy[B](f: A => B)(implicit num: Numeric[B]): B = {
      as.foldLeft(num.zero)((acc, a) => num.plus(acc, f(a)))
    }
  }
}

Submission Info

Submission Time
Task C - Sequence
User yakamoto
Language Scala (2.11.7)
Score 300
Code Size 3552 Byte
Status AC
Exec Time 526 ms
Memory 42572 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 300 / 300
Status
AC × 3
AC × 18
Set Name Test Cases
Sample 00-00.txt, 00-01.txt, 00-02.txt
All 00-00.txt, 00-01.txt, 00-02.txt, 01-00.txt, 01-01.txt, 01-02.txt, 01-03.txt, 01-04.txt, 01-05.txt, 01-06.txt, 01-07.txt, 01-08.txt, 01-09.txt, 01-10.txt, 01-11.txt, 01-12.txt, 01-13.txt, 01-14.txt
Case Name Status Exec Time Memory
00-00.txt AC 322 ms 25416 KB
00-01.txt AC 316 ms 25420 KB
00-02.txt AC 320 ms 25380 KB
01-00.txt AC 483 ms 38580 KB
01-01.txt AC 494 ms 42564 KB
01-02.txt AC 489 ms 39604 KB
01-03.txt AC 486 ms 41760 KB
01-04.txt AC 484 ms 41764 KB
01-05.txt AC 526 ms 37636 KB
01-06.txt AC 492 ms 40024 KB
01-07.txt AC 494 ms 38232 KB
01-08.txt AC 489 ms 36712 KB
01-09.txt AC 492 ms 39812 KB
01-10.txt AC 514 ms 37148 KB
01-11.txt AC 507 ms 39656 KB
01-12.txt AC 514 ms 39912 KB
01-13.txt AC 501 ms 39848 KB
01-14.txt AC 519 ms 42572 KB