結果
問題 | No.1124 Earthquake Safety |
ユーザー | yakamoto |
提出日時 | 2020-07-22 23:17:22 |
言語 | Kotlin (1.9.23) |
結果 |
AC
|
実行時間 | 1,052 ms / 3,000 ms |
コード長 | 4,445 bytes |
コンパイル時間 | 22,757 ms |
コンパイル使用メモリ | 437,228 KB |
実行使用メモリ | 84,000 KB |
最終ジャッジ日時 | 2023-09-05 04:22:58 |
合計ジャッジ時間 | 69,476 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 272 ms
50,212 KB |
testcase_01 | AC | 274 ms
50,148 KB |
testcase_02 | AC | 272 ms
50,664 KB |
testcase_03 | AC | 273 ms
50,044 KB |
testcase_04 | AC | 272 ms
50,228 KB |
testcase_05 | AC | 277 ms
50,152 KB |
testcase_06 | AC | 293 ms
52,524 KB |
testcase_07 | AC | 418 ms
57,492 KB |
testcase_08 | AC | 680 ms
65,828 KB |
testcase_09 | AC | 1,020 ms
82,832 KB |
testcase_10 | AC | 826 ms
82,524 KB |
testcase_11 | AC | 273 ms
50,176 KB |
testcase_12 | AC | 271 ms
50,348 KB |
testcase_13 | AC | 272 ms
50,188 KB |
testcase_14 | AC | 828 ms
83,500 KB |
testcase_15 | AC | 840 ms
83,196 KB |
testcase_16 | AC | 878 ms
83,212 KB |
testcase_17 | AC | 860 ms
83,040 KB |
testcase_18 | AC | 859 ms
83,116 KB |
testcase_19 | AC | 844 ms
82,932 KB |
testcase_20 | AC | 872 ms
83,012 KB |
testcase_21 | AC | 974 ms
83,428 KB |
testcase_22 | AC | 867 ms
83,044 KB |
testcase_23 | AC | 855 ms
82,848 KB |
testcase_24 | AC | 841 ms
83,024 KB |
testcase_25 | AC | 842 ms
82,816 KB |
testcase_26 | AC | 872 ms
83,272 KB |
testcase_27 | AC | 900 ms
83,004 KB |
testcase_28 | AC | 827 ms
82,968 KB |
testcase_29 | AC | 868 ms
83,164 KB |
testcase_30 | AC | 835 ms
83,048 KB |
testcase_31 | AC | 851 ms
83,312 KB |
testcase_32 | AC | 826 ms
82,916 KB |
testcase_33 | AC | 846 ms
82,192 KB |
testcase_34 | AC | 833 ms
82,156 KB |
testcase_35 | AC | 854 ms
83,304 KB |
testcase_36 | AC | 896 ms
82,840 KB |
testcase_37 | AC | 887 ms
83,420 KB |
testcase_38 | AC | 867 ms
83,212 KB |
testcase_39 | AC | 875 ms
83,064 KB |
testcase_40 | AC | 875 ms
83,316 KB |
testcase_41 | AC | 843 ms
83,232 KB |
testcase_42 | AC | 1,052 ms
84,000 KB |
testcase_43 | AC | 857 ms
83,260 KB |
testcase_44 | AC | 819 ms
83,388 KB |
testcase_45 | AC | 857 ms
83,540 KB |
testcase_46 | AC | 831 ms
83,416 KB |
testcase_47 | AC | 870 ms
83,580 KB |
testcase_48 | AC | 859 ms
83,320 KB |
testcase_49 | AC | 840 ms
83,256 KB |
testcase_50 | AC | 274 ms
50,184 KB |
testcase_51 | AC | 274 ms
50,060 KB |
testcase_52 | AC | 271 ms
50,256 KB |
testcase_53 | AC | 271 ms
50,252 KB |
testcase_54 | AC | 269 ms
50,396 KB |
testcase_55 | AC | 270 ms
50,128 KB |
testcase_56 | AC | 267 ms
49,972 KB |
testcase_57 | AC | 277 ms
50,232 KB |
ソースコード
import java.io.BufferedReader import java.io.InputStream import java.io.InputStreamReader import java.util.* import kotlin.math.abs import kotlin.math.max import kotlin.math.min val MOD = 1_000_000_007L /** * (dist, parent, queue) */ fun traceBfs(g: Array<IntArray>, rt: Int = 0): Array<IntArray> { val n = g.size val q = IntArray(n) val d = IntArray(n) val p = IntArray(n){-2} var cur = 0 var last = 1 p[rt] = -1 q[0] = rt d[rt] = 0 while (cur < last) { val v = q[cur++] for (u in g[v]) { if (p[u] == -2) { p[u] = v q[last++] = u d[u] = d[v] + 1 } } } return arrayOf(d, p, q) } fun packUGraph(n: Int, from: IntArray, to: IntArray): Array<IntArray> { val p = IntArray(n) val m = from.size for (i in 0 until m) { ++p[from[i]] ++p[to[i]] } val g = Array(n){IntArray(p[it])} for (i in 0 until m) { g[from[i]][--p[from[i]]] = to[i] g[to[i]][--p[to[i]]] = from[i] } return g } class Solver(stream: InputStream, private val out: java.io.PrintWriter) { fun solve() { val N = ni() val (from, to) = na2(N - 1, -1) val g = packUGraph(N, from, to) val (_, par, que) = traceBfs(g) val pow2 = LongArray(N + 1) pow2[0] = 1 for (i in 1 until pow2.size) { pow2[i] = pow2[i - 1] * 2 % MOD } var ans = 0L val dpc = IntArray(N){1} val dp = Array(4){LongArray(N){1} } for (i in N - 1 downTo 0) { val u = que[i] for (v in g[u]) { if (v == par[u]) continue dp[3][u] = dp[3][u] * dp[0][v] % MOD + dp[3][v] * dp[0][u] % MOD + 3 * dp[2][v] * dp[1][u] % MOD + 3 * dp[2][u] * dp[1][v] % MOD + dp[3][u] * dp[0][v] % MOD dp[3][u] %= MOD dp[2][u] = dp[2][u] * dp[0][v] % MOD + dp[2][v] * dp[0][u] % MOD + 2 * dp[1][v] * dp[1][u] % MOD + dp[2][u] * dp[0][v] % MOD dp[2][u] %= MOD dp[1][u] = dp[1][u] * dp[0][v] % MOD + dp[1][v] * dp[0][u] % MOD + dp[1][u] * dp[0][v] % MOD dp[1][u] %= MOD dp[0][u] = dp[0][u] * dp[0][v] * 2 % MOD dpc[u] += dpc[v] debug{"$u->$v"} } debug{"$u"} debugDim(dp) val outNodes = N - dpc[u] ans = (ans + dp[3][u] * pow2[max(0, outNodes - 1)]) % MOD } out.println(ans) } private val isDebug = try { // なんか本番でエラーでる System.getenv("MY_DEBUG") != null } catch (t: Throwable) { false } private var tokenizer: StringTokenizer? = null private val reader = BufferedReader(InputStreamReader(stream), 32768) private fun next(): String { while (tokenizer == null || !tokenizer!!.hasMoreTokens()) { tokenizer = StringTokenizer(reader.readLine()) } return tokenizer!!.nextToken() } private fun ni() = next().toInt() private fun nl() = next().toLong() private fun ns() = next() private fun na(n: Int, offset: Int = 0): IntArray { return map(n) { ni() + offset } } private fun nal(n: Int, offset: Int = 0): LongArray { val res = LongArray(n) for (i in 0 until n) { res[i] = nl() + offset } return res } private fun na2(n: Int, offset: Int = 0): Array<IntArray> { val a = Array(2){IntArray(n)} for (i in 0 until n) { for (e in a) { e[i] = ni() + offset } } return a } private inline fun map(n: Int, f: (Int) -> Int): IntArray { val res = IntArray(n) for (i in 0 until n) { res[i] = f(i) } return res } private inline fun debug(msg: () -> String) { if (isDebug) System.err.println(msg()) } private fun debug(a: LongArray) { debug { a.joinToString(" ") } } private fun debug(a: IntArray) { debug { a.joinToString(" ") } } private fun debug(a: BooleanArray) { debug { a.map { if (it) 1 else 0 }.joinToString("") } } private fun debugDim(A: Array<LongArray>) { if (isDebug) { for (a in A) { debug(a) } } } private fun debugDim(A: Array<IntArray>) { if (isDebug) { for (a in A) { debug(a) } } } /** * 勝手にimport消されるのを防ぎたい */ private fun hoge() { min(1, 2) max(1, 2) abs(-10) } } fun main() { val out = java.io.PrintWriter(System.out) Solver(System.`in`, out).solve() out.flush() }