P3085 [USACO13OPEN] Yin and Yang G

题目描述

农夫约翰正计划在农场进行晨间散步。农场结构如同一棵树: 拥有 $N$ 个谷仓$(1 \leq N \leq 100,000)$,这些谷仓通过 $N-1$ 条边相连, 使得他可以从任意谷仓到达其他任何谷仓。农夫约翰希望选择一条 起始于不同谷仓的路径,且不重复经过任何边。他担心路径可能过长, 因此还想在这条路径上选择一个"休息站"谷仓(该谷仓不能是起点或终点)。 每条边上都有一群牛,要么是夏洛莱牛(白毛),要么是安格斯牛(黑毛)。 作为智者,农夫约翰希望平衡影响他散步的阴阳之力。为此,他需要选择这样一条路径: 从起点到休息站经过的夏洛莱牛群与安格斯牛群数量相等, 同时从休息站到终点经过的两种牛群数量也相等。 农夫约翰想知道有多少条符合上述"平衡"条件的路径可供选择。 两条路径仅当其边集不同时才被视为不同;即使某条路径上有多个 符合条件的休息站位置,该路径也应只被计数一次。 请帮助计算农夫约翰可以选择的路径数量!

输入格式

* 第 $1$ 行: 整数 $N$ * 第 $2$ ~ $N$ 行: 三个整数 $a_i$ , $b_i$ 和 $t_i$,表示边 $i$ 连接的两个谷仓。 * $t_i$为 $0$ 表示该边上的牛群是夏洛莱牛,$1$ 表示是安格斯牛。

输出格式

* 第 $1$ 行: 一个整数,表示农夫约翰可以选择的可能路径数量

说明/提示

共有 $7$ 个谷仓和 $6$ 条边。连接 $1-2$、$2-4$ 和 $2-5$ 的边上有夏洛莱牛群。 长度为 $2$ 的路径上不可能存在合适的休息站,因此我们只能考虑 长度为 $4$ 的路径。唯一符合条件的路径是 $3-1-2-5-7$,休息站设在 $2$ 号谷仓。