結果

問題 No.872 All Tree Path
ユーザー te-sh
提出日時 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

ソースコード

diff #
プレゼンテーションモードにする

def solve(io)
n = io.get
u, 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)
end
t = 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] = wi
end
ans = 0_i64
(1...n).each do |u|
p = t.descendant_size[u]
ans += e[u] * p * (n - p)
end
io.put ans * 2
end
class ProconIO
def initialize(@ins : IO = STDIN, @outs : IO = STDOUT)
@buf = IO::Memory.new("")
end
def get(k : T.class = Int32) forall T
get_v(k)
end
macro 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 %}
end
define_get
macro define_getn
{% for i in (2..9) %}
def get{{i}}(k : T.class = Int32) forall T
get({{ *(1..i).map { "k".id } }})
end
{% end %}
end
define_getn
def get_a(n : Int, k : T.class = Int32) forall T
Array.new(n) { get_v(k) }
end
def get_c(n : Int, k : T.class = Int32) forall T
get_a(n, k)
end
macro 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 %}
end
define_get_c
macro define_getn_c
{% for i in (2..9) %}
def get{{i}}_c(n : Int, k : T.class = Int32) forall T
get_c(n, {{ *(1..i).map { "k".id } }})
end
{% end %}
end
define_getn_c
def get_m(r : Int, c : Int, k : T.class = Int32) forall T
Array.new(r) { get_a(c, k) }
end
macro 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.puts
end
{% end %}
end
define_put
def put_e(*vs)
put(*vs)
exit
end
def put_f(*vs)
put(*vs)
@outs.flush
end
private def get_v(k : Int32.class); get_token.to_i32; end
private def get_v(k : Int64.class); get_token.to_i64; end
private def get_v(k : UInt32.class); get_token.to_u32; end
private def get_v(k : UInt64.class); get_token.to_u64; end
private def get_v(k : Float64.class); get_token.to_f64; end
private def get_v(k : String.class); get_token; end
private def get_token
loop do
token = @buf.gets(' ', chomp: true)
break token unless token.nil?
@buf = IO::Memory.new(@ins.read_line)
end
end
private def print_v(v, dlimiter)
@outs << v
end
private def print_v(v : Enumerable, delimiter)
v.each_with_index do |e, i|
@outs << e
@outs << delimiter if i < v.size - 1
end
end
end
struct Int
def cdiv(b : Int)
(self + b - 1) // b
end
def bit?(i : Int)
bit(i) == 1
end
def set_bit(i : Int)
self | (self.class.new(1) << i)
end
def 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 : Int32
x = self < 0 ? ~self : self
if x.is_a?(Int::Primitive)
Int32.new(sizeof(self) * 8 - x.leading_zeros_count)
else
to_s(2).size
end
end
{% end %}
end
struct Float64
def near_zero?
self.abs < EPSILON
end
end
struct Number
{% if compare_versions(env("CRYSTAL_VERSION") || "0.0.0", "1.1.0") < 0 %}
def zero?
self == 0
end
def positive?
self > 0
end
def negative?
self < 0
end
{% end %}
{% if compare_versions(env("CRYSTAL_VERSION") || "0.0.0", "0.36.0") < 0 %}
def self.additive_identity
zero
end
def self.multiplicative_identity
new(1)
end
{% end %}
end
class Array
macro 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 %}
end
end
module 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 < 0
return value if value < 2
res = value.class.zero
bit = res.succ << (res.leading_zeros_count - 2)
bit >>= value.leading_zeros_count & ~0x3
while (bit != 0)
if value >= res + bit
value -= res + bit
res = (res >> 1) + bit
else
res >>= 1
end
bit >>= 2
end
res
end
{% end %}
end
macro min_u(a, b)
{{a}} = Math.min({{a}}, {{b}})
end
macro max_u(a, b)
{{a}} = Math.max({{a}}, {{b}})
end
macro zip(a, *b, &block)
{{a}}.zip({{*b}}) {{block}}
end
require "bit_array"
class Graph
alias Node = Int32
def initialize(@size : Node)
@g = Array(Array(Node)).new(@size) { [] of Node }
end
getter size : Int32
delegate :[], to: @g
def add_edge(u : Node, v : Node)
@g[u] << v
end
def add_edge_b(u : Node, v : Node)
@g[u] << v
@g[v] << u
end
def bfs(u : Node)
b = BitArray.new(@size)
yield u, -1
b[u] = true
q = Deque.new([u])
until q.empty?
v = q.shift
@g[v].each do |w|
next if b[w]
yield w, v
b[w] = true
q.push(w)
end
end
end
end
class GraphW(T)
alias Node = Int32
struct Edge(T)
def initialize(@src : Node, @dst : Node, @wt : T)
end
getter src : Node, dst : Node
getter wt : T
end
def initialize(@size : Node, @inf = 10**9)
@g = Array(Array(Edge(T))).new(@size) { [] of Edge(T) }
end
getter size : Int32
getter inf : T
delegate :[], to: @g
def add_edge(u : Node, v : Node, wt : T)
@g[u] << Edge.new(u, v, wt)
end
def add_edge_b(u : Node, v : Node, wt : T)
@g[u] << Edge.new(u, v, wt)
@g[v] << Edge.new(v, u, wt)
end
end
class GraphM(T)
alias Node = Int32
def initialize(@size : Int32, @inf = 10**9)
@g = Array.new_md(@size, @size, @inf)
@size.times do |i|
@g[i][i] = T.zero
end
end
getter size : Int32
getter inf : T
delegate :[], to: @g
def add_edge(u : Node, v : Node, wt : T)
@g[u][v] = wt
end
def add_edge_b(u : Node, v : Node, wt : T)
@g[u][v] = @g[v][u] = wt
end
@g : Array(Array(T))
end
class Tree
alias Node = Graph::Node
def 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 Node
s = [{@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] << v
s.push({v, u})
end
end
end
@descendant_size = Array.new(size, 1)
@dfs_order.reverse_each do |u|
@descendant_size[@parent[u]] += @descendant_size[u] if u != @root
end
end
delegate size, to: @g
getter root : Node
getter dfs_order : Array(Node)
getter parent : Array(Node)
getter children : Array(Array(Node))
getter depth : Array(Int32)
getter descendant_size : Array(Int32)
end
class Graph
def tree(root)
Tree.new(self, root)
end
end
solve(ProconIO.new)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0