问题分析
树上每条简单路径的共鸣能量由路径上结点的能量值决定:
因此,只有路径上的结点能量值完全相同时,共鸣能量才为 3 或 5;否则为 15。
给定一棵树,每个结点都带有一个能量值,该值只可能是 3 或 5。对于树上的一条简单路径,定义该路径的共鸣能量为:若路径上所有结点的能量值全为 3,则共鸣能量为 3;若全为 5,则共鸣能量为 5;若路径上同时出现 3 和 5,则共鸣能量为 15。
试求树上所有简单路径的共鸣能量之和。
结点数量 n 不超过 105,所有结点的能量值只可能是 3 或 5。
第一行包含一个整数 n,表示树的结点数量。 第二行包含 n 个整数,第 i 个整数 ai 表示第 i 个结点的能量值,保证 ai 只可能是 3 或 5。
By signing up a CodeFun2000 universal account, you can submit code and join discussions in all online judging services provided by us.