結果
問題 | No.872 All Tree Path |
ユーザー |
|
提出日時 | 2021-07-30 14:38:51 |
言語 | Crystal (1.14.0) |
結果 |
AC
|
実行時間 | 199 ms / 3,000 ms |
コード長 | 7,831 bytes |
コンパイル時間 | 15,803 ms |
コンパイル使用メモリ | 307,004 KB |
実行使用メモリ | 41,984 KB |
最終ジャッジ日時 | 2024-09-15 10:13:44 |
合計ジャッジ時間 | 19,112 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
コンパイルメッセージ
In Main.cr:37:19 37 | def get({{ *(1..i).map { |j| "k#{j}".id } }}) ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:38:15 38 | { {{ *(1..i).map { |j| "get(k#{j})".id } }} } ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:37:19 37 | def get({{ *(1..i).map { |j| "k#{j}".id } }}) ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:38:15 38 | { {{ *(1..i).map { |j| "get(k#{j})".id } }} } ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:37:19 37 | def get({{ *(1..i).map { |j| "k#{j}".id } }}) ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:38:15 38 | { {{ *(1..i).map { |j| "get(k#{j})".id } }} } ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:37:19 37 | def get({{ *(1..i).map { |j| "k#{j}".id } }}) ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:38:15 38 | { {{ *(1..i).map { |j| "get(k#{j})".id } }} } ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:37:19 37 | def get({{ *(1..i).map { |j| "k#{j}".id } }}) ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:38:15 38 | { {{ *(1..i).map { |j| "get(k#{j})".id } }} } ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:37:19 37 | def get({{ *(1..i).map { |j| "k#{j}".id } }}) ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:38:15 38 | { {{ *(1..i).map { |j| "get(k#{j})".id } }} } ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:37:19 37 | def get({{ *(1..i).map { |j| "k#{j}".id } }}) ^ Warning: Deprecated use of splat operator. Use `#splat` instead In Main.cr:38:15 38
ソースコード
def solve(io)n = io.getu, v, w = io.get_c(n - 1, Int32, Int32, Int64); u.map!(&.- 1); v.map!(&.- 1)g = Graph.new(n)zip(u, v).each do |ui, vi|g.add_edge_b(ui, vi)endt = g.tree(0)e = Array.new(n, 0_i64)zip(u, v, w).each do |ui, vi, wi|e[t.depth[ui] > t.depth[vi] ? ui : vi] = wiendans = 0_i64(1...n).each do |u|p = t.descendant_size[u]ans += e[u] * p * (n - p)endio.put ans * 2endclass ProconIOdef initialize(@ins : IO = STDIN, @outs : IO = STDOUT)@buf = IO::Memory.new("")enddef get(k : T.class = Int32) forall Tget_v(k)endmacro define_get{% for i in (2..9) %}def get({{ *(1..i).map { |j| "k#{j}".id } }}){ {{ *(1..i).map { |j| "get(k#{j})".id } }} }end{% end %}enddefine_getmacro define_getn{% for i in (2..9) %}def get{{i}}(k : T.class = Int32) forall Tget({{ *(1..i).map { "k".id } }})end{% end %}enddefine_getndef get_a(n : Int, k : T.class = Int32) forall TArray.new(n) { get_v(k) }enddef get_c(n : Int, k : T.class = Int32) forall Tget_a(n, k)endmacro define_get_c{% for i in (2..9) %}def get_c(n : Int, {{ *(1..i).map { |j| "k#{j}".id } }})a = Array.new(n) { get({{ *(1..i).map { |j| "k#{j}".id } }}) }{ {{ *(1..i).map { |j| "a.map { |e| e[#{j-1}] }".id } }} }end{% end %}enddefine_get_cmacro define_getn_c{% for i in (2..9) %}def get{{i}}_c(n : Int, k : T.class = Int32) forall Tget_c(n, {{ *(1..i).map { "k".id } }})end{% end %}enddefine_getn_cdef get_m(r : Int, c : Int, k : T.class = Int32) forall TArray.new(r) { get_a(c, k) }endmacro define_put{% for i in (1..9) %}def put({{ *(1..i).map { |j| "v#{j}".id } }}, *, delimiter = " "){% for j in (1..i) %}print_v(v{{j}}, delimiter){% if j < i %}@outs << delimiter{% end %}{% end %}@outs.putsend{% end %}enddefine_putdef put_e(*vs)put(*vs)exitenddef put_f(*vs)put(*vs)@outs.flushendprivate def get_v(k : Int32.class); get_token.to_i32; endprivate def get_v(k : Int64.class); get_token.to_i64; endprivate def get_v(k : UInt32.class); get_token.to_u32; endprivate def get_v(k : UInt64.class); get_token.to_u64; endprivate def get_v(k : Float64.class); get_token.to_f64; endprivate def get_v(k : String.class); get_token; endprivate def get_tokenloop dotoken = @buf.gets(' ', chomp: true)break token unless token.nil?@buf = IO::Memory.new(@ins.read_line)endendprivate def print_v(v, dlimiter)@outs << vendprivate def print_v(v : Enumerable, delimiter)v.each_with_index do |e, i|@outs << e@outs << delimiter if i < v.size - 1endendendstruct Intdef cdiv(b : Int)(self + b - 1) // benddef bit?(i : Int)bit(i) == 1enddef set_bit(i : Int)self | (self.class.new(1) << i)enddef reset_bit(i : Int)self & ~(self.class.new(1) << i)end{% if compare_versions(env("CRYSTAL_VERSION") || "0.0.0", "0.34.0") < 0 %}def bit_length : Int32x = self < 0 ? ~self : selfif x.is_a?(Int::Primitive)Int32.new(sizeof(self) * 8 - x.leading_zeros_count)elseto_s(2).sizeendend{% end %}endstruct Float64def near_zero?self.abs < EPSILONendendstruct Number{% if compare_versions(env("CRYSTAL_VERSION") || "0.0.0", "1.1.0") < 0 %}def zero?self == 0enddef positive?self > 0enddef negative?self < 0end{% end %}{% if compare_versions(env("CRYSTAL_VERSION") || "0.0.0", "0.36.0") < 0 %}def self.additive_identityzeroenddef self.multiplicative_identitynew(1)end{% end %}endclass Arraymacro new_md(*args, &block){% if !block %}{% for arg, i in args[0...-2] %}Array.new({{arg}}) {{% end %}Array.new({{args[-2]}}, {{args[-1]}}){% for arg in args[0...-2] %}}{% end %}{% else %}{% for arg, i in args %}Array.new({{arg}}) { |_i{{i}}|{% end %}{% for block_arg, i in block.args %}{{block_arg}} = _i{{i}}{% end %}{{block.body}}{% for arg in args %}}{% end %}{% end %}endendmodule Math{% if compare_versions(env("CRYSTAL_VERSION") || "0.0.0", "1.2.0") < 0 %}def isqrt(value : Int::Primitive)raise ArgumentError.new "Input must be non-negative integer" if value < 0return value if value < 2res = value.class.zerobit = res.succ << (res.leading_zeros_count - 2)bit >>= value.leading_zeros_count & ~0x3while (bit != 0)if value >= res + bitvalue -= res + bitres = (res >> 1) + bitelseres >>= 1endbit >>= 2endresend{% end %}endmacro min_u(a, b){{a}} = Math.min({{a}}, {{b}})endmacro max_u(a, b){{a}} = Math.max({{a}}, {{b}})endmacro zip(a, *b, &block){{a}}.zip({{*b}}) {{block}}endrequire "bit_array"class Graphalias Node = Int32def initialize(@size : Node)@g = Array(Array(Node)).new(@size) { [] of Node }endgetter size : Int32delegate :[], to: @gdef add_edge(u : Node, v : Node)@g[u] << venddef add_edge_b(u : Node, v : Node)@g[u] << v@g[v] << uenddef bfs(u : Node)b = BitArray.new(@size)yield u, -1b[u] = trueq = Deque.new([u])until q.empty?v = q.shift@g[v].each do |w|next if b[w]yield w, vb[w] = trueq.push(w)endendendendclass GraphW(T)alias Node = Int32struct Edge(T)def initialize(@src : Node, @dst : Node, @wt : T)endgetter src : Node, dst : Nodegetter wt : Tenddef initialize(@size : Node, @inf = 10**9)@g = Array(Array(Edge(T))).new(@size) { [] of Edge(T) }endgetter size : Int32getter inf : Tdelegate :[], to: @gdef add_edge(u : Node, v : Node, wt : T)@g[u] << Edge.new(u, v, wt)enddef add_edge_b(u : Node, v : Node, wt : T)@g[u] << Edge.new(u, v, wt)@g[v] << Edge.new(v, u, wt)endendclass GraphM(T)alias Node = Int32def initialize(@size : Int32, @inf = 10**9)@g = Array.new_md(@size, @size, @inf)@size.times do |i|@g[i][i] = T.zeroendendgetter size : Int32getter inf : Tdelegate :[], to: @gdef add_edge(u : Node, v : Node, wt : T)@g[u][v] = wtenddef add_edge_b(u : Node, v : Node, wt : T)@g[u][v] = @g[v][u] = wtend@g : Array(Array(T))endclass Treealias Node = Graph::Nodedef initialize(@g : Graph, @root : Node)size = @g.size@parent = Array.new(size, 0)@children = Array.new(size) { [] of Node }@depth = Array.new(size, -1)@dfs_order = [] of Nodes = [{@root, @root}]until s.empty?u, p = s.pop@parent[u] = p@depth[u] = @depth[p] + 1@dfs_order << u@g[u].each do |v|if v != p@children[u] << vs.push({v, u})endendend@descendant_size = Array.new(size, 1)@dfs_order.reverse_each do |u|@descendant_size[@parent[u]] += @descendant_size[u] if u != @rootendenddelegate size, to: @ggetter root : Nodegetter dfs_order : Array(Node)getter parent : Array(Node)getter children : Array(Array(Node))getter depth : Array(Int32)getter descendant_size : Array(Int32)endclass Graphdef tree(root)Tree.new(self, root)endendsolve(ProconIO.new)