結果

問題 No.848 なかよし旅行
ユーザー yakamotoyakamoto
提出日時 2019-07-12 01:32:52
言語 Scala(Beta)
(3.4.0)
結果
AC  
実行時間 1,328 ms / 2,000 ms
コード長 5,576 bytes
コンパイル時間 12,680 ms
コンパイル使用メモリ 279,092 KB
実行使用メモリ 81,772 KB
最終ジャッジ日時 2024-04-25 13:57:53
合計ジャッジ時間 43,778 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,239 ms
77,516 KB
testcase_01 AC 784 ms
63,900 KB
testcase_02 AC 825 ms
63,976 KB
testcase_03 AC 787 ms
64,084 KB
testcase_04 AC 794 ms
63,992 KB
testcase_05 AC 802 ms
64,064 KB
testcase_06 AC 824 ms
64,188 KB
testcase_07 AC 805 ms
64,100 KB
testcase_08 AC 835 ms
64,212 KB
testcase_09 AC 851 ms
64,360 KB
testcase_10 AC 811 ms
64,192 KB
testcase_11 AC 1,080 ms
67,164 KB
testcase_12 AC 1,096 ms
67,204 KB
testcase_13 AC 1,135 ms
69,192 KB
testcase_14 AC 1,066 ms
65,176 KB
testcase_15 AC 1,328 ms
81,772 KB
testcase_16 AC 1,198 ms
70,428 KB
testcase_17 AC 1,047 ms
69,044 KB
testcase_18 AC 1,028 ms
66,960 KB
testcase_19 AC 1,032 ms
64,776 KB
testcase_20 AC 987 ms
64,588 KB
testcase_21 AC 1,070 ms
69,008 KB
testcase_22 AC 1,142 ms
70,384 KB
testcase_23 AC 967 ms
64,636 KB
testcase_24 AC 813 ms
64,016 KB
testcase_25 AC 1,165 ms
70,256 KB
testcase_26 AC 797 ms
63,984 KB
testcase_27 AC 795 ms
63,972 KB
testcase_28 AC 785 ms
63,972 KB
testcase_29 AC 814 ms
64,108 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

object Main {
  import java.io.{BufferedReader, InputStream, InputStreamReader}
  import java.util.StringTokenizer
  import scala.reflect.ClassTag

  def main(args: Array[String]): Unit = {
    val out = new java.io.PrintWriter(System.out)
    new Main(out, new InputReader(System.in)).solve()
    out.flush()
  }

  private val oj = System.getenv("ATCODER_DEBUG") == null
  def DEBUG(f: => Unit): Unit = {
    if (!oj){ f }
  }
  def debug(as: Array[Boolean]): Unit = if (!oj){ debug(as.map(x => if(x) "1" else "0").mkString) }
  def debug(as: Array[Int]): Unit = if (!oj){ debug(as.mkString(" ")) }
  def debug(as: Array[Long]): Unit =if (!oj){ debug(as.mkString(" ")) }
  def debugDim(m: Array[Array[Long]]): Unit = if (!oj){
    REP(m.length) { i =>
      debug(m(i))
    }
  }
  def debugDimFlip(m: Array[Array[Long]]): Unit = if (!oj){
    REP(m(0).length) { j =>
      REP(m.length) { i =>
        System.err.print(m(i)(j))
        System.err.print(" ")
      }
      System.err.println()
    }
  }
  def debug(s: => String): Unit = DEBUG {
    System.err.println(s)
  }
  def debugL(num: => Long): Unit = DEBUG {
    System.err.println(num)
  }

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

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

    def nextInt(): Int = Integer.parseInt(next())
    def nextLong(): Long = java.lang.Long.parseLong(next())
    def nextChar(): Char = next().charAt(0)

    def ni(): Int = nextInt()
    def nl(): Long = nextLong()
    def nc(): Char = nextChar()
    def ns(): String = next()
    def ns(n: Int): Array[Char] = ns().toCharArray
    def na(n: Int, offset: Int = 0): Array[Int] = map(n)(_ => ni() + offset)
    def na2(n: Int, offset: Int = 0): (Array[Int], Array[Int]) = {
      val A1, A2 = Array.ofDim[Int](n)
      REP(n) { i =>
        A1(i) = ni() + offset
        A2(i) = ni() + offset
      }
      (A1, A2)
    }
    def nm(n: Int, m: Int): Array[Array[Int]] = {
      val A = Array.ofDim[Int](n, m)
      REP(n) { i =>
        REP(m) { j =>
          A(i)(j) = ni()
        }
      }
      A
    }
    def nal(n: Int): Array[Long] = map(n)(_ => nl())
    def nm_c(n: Int, m: Int): Array[Array[Char]] = map(n) (_ => ns(m))
  }

  def REP(n: Int, offset: Int = 0)(f: Int => Unit): Unit = {
    var i = offset
    val N = n + offset
    while(i < N) { f(i); i += 1 }
  }
  def REP_r(n: Int, offset: Int = 0)(f: Int => Unit): Unit = {
    var i = n - 1 + offset
    while(i >= offset) { f(i); i -= 1 }
  }
  def TO(from: Int, to: Int)(f: Int => Unit): Unit = {
    REP(to - from + 1, from) { i =>
      f(i)
    }
  }
  def map[@specialized A: ClassTag](n: Int, offset: Int = 0)(f: Int => A): Array[A] = {
    val res = Array.ofDim[A](n)
    REP(n)(i => res(i) = f(i + offset))
    res
  }

  def sumL(as: Array[Int]): Long = {
    var s = 0L
    REP(as.length)(i => s += as(i))
    s
  }
  def cumSum(as: Array[Int]): Array[Long] = {
    val cum = Array.ofDim[Long](as.length + 1)
    REP(as.length) { i =>
      cum(i + 1) = cum(i) + as(i)
    }
    cum
  }
}

class Main(out: java.io.PrintWriter, sc: Main.InputReader) {
  import sc._
  import Main._
  import java.util.Arrays.sort

  import scala.collection.mutable
  import math.{abs, max, min}
  import mutable.ArrayBuffer

  // toIntとか+7とかするならvalにしろ
  @inline private def MOD = 1000000007

  def solve(): Unit = {
    val N, M = ni()
    val P, Q = ni() - 1
    val T = ni()
    val from, to, w = Array.ofDim[Int](M)
    REP(M) { i =>
      val a, b = ni() - 1
      val c = ni()
      from(i) = a
      to(i) = b
      w(i) = c
    }
    val g = packWUGraph(N, from, to, w)
    val D0 = dijk(g, 0)
    val Dp = dijk(g, P)
    val Dq = dijk(g, Q)

    var ans = -1L
    if (D0(P) + D0(Q) + Dp(Q) <= T) {
      ans = T
    } else {
      REP(N) { i =>
        REP(N) { j =>
          // 0 -> i -> P/Q -> j -> 0
          val separeted = max(Dp(i) + Dp(j), Dq(i) + Dq(j))
          val t = D0(i) + separeted + D0(j)
          if (t <= T) ans = max(ans, T - separeted)
        }
      }
    }

    out.println(ans)
  }

  case class Edge(to: Int, weight: Int)
  type WUGraph = Array[Array[Edge]]
  def packWUGraph(n: Int, from: Array[Int], to: Array[Int], w: Array[Int]): WUGraph = {
    val g = new Array[Array[Edge]](n)
    val p = new Array[Int](n)
    val m = from.length
    REP(m)(i => p(from(i)) += 1)
    REP(m)(i => p(to(i)) += 1)
    REP(n)(i => g(i) = new Array(p(i)))
    REP(m) { i =>
      p(from(i)) -= 1
      g(from(i))(p(from(i))) = Edge(to(i), w(i))
      p(to(i)) -= 1
      g(to(i))(p(to(i))) = Edge(from(i), w(i))
    }
    g
  }

  case class Visit(node: Int, cost: Long) extends Comparable[Visit] {
    override def compareTo(o: Visit): Int = java.lang.Long.compare(cost, o.cost)
  }
  val queue = new java.util.PriorityQueue[Visit](1e5.toInt) // 使い回す
  def dijk(g: WUGraph, start: Int): Array[Long] = {
    val d = Array.fill[Long](g.length)(Long.MaxValue / 2)
    d(start) = 0
    queue.add(Visit(start, 0))

    while(!queue.isEmpty) {
      val v = queue.poll()
      if (d(v.node) == v.cost) {
        REP(g(v.node).length) { i =>
          val e = g(v.node)(i)
          val next = v.cost + e.weight
          if (d(e.to) > next) {
            d(e.to) = next
            queue.add(Visit(e.to, next))
          }
        }
      }
    }

    d
  }
}
0