#L6119. 「2017 山东二轮集训 Day7」国王

    ID: 6089 传统题 1000ms 256MiB 尝试: 1 已通过: 1 难度: 10 上传者: 标签>数据结构差分前缀和平衡树树论点分治树上路径统计

「2017 山东二轮集训 Day7」国王

题目描述

在某个神奇的大陆上,有一个国家,这片大陆的所有城市间的道路网可以看作是一棵树,每个城市要么是工业城市(记为 11),要么是农业城市(记为 00)。

这个国家的人认为一条路径是 exciting 的,当且仅当这条路径上的工业城市和农业城市数目相等。

现在国王想把城市分给他的两个儿子,大儿子想知道,他选择一段标号连续的城市作为自己的领地,并把剩下的给弟弟,能够满足两端都是自己城市的 exciting 路径两端都是弟弟的城市的 exciting 路径数目多的方案数。


输入格式

第一行一个正整数 nn
第二行 nn 个整数依次描述城市的性质,11 为工业,00 为农业。
接下来 n1n - 1 行每行两个正整数描述一条道路。


输出格式

输出一个整数表示答案。


样例

输入

5
1 0 1 0 1
1 2
1 3
2 4
2 5

输出

5

数据范围与提示

n100000n \leq 100000