170. 权值不等的路径方案
内存限制:256 MB
时间限制:1.000 S
题目描述
小红拿到了一个有根树,根节点为1号节点,每个节点到其每个孩子有条有向边。小红想取一条路径,满足路径上所有节点的权值都不相等。小红想知道,自己有多少种选择方案?
输入
第一行输入一个正整数 n,代表节点的数量。
第二行输入 n - 1 个正整数 ai,代表 2 号节点到 n 号节点每个节点的父亲编号。
第三行输入 n 个正整数 vi,代表 1 号节点到 n 号节点每个节点的权值。
输出
输出一个整数,代表方案数。
样例输入 复制
5
1 1 1 2
1 1 2 2 2
样例输出 复制
8
提示
路径上有一个节点,有 5 种取法。路径上有两个节点的取法有: 1-3、1-4、2-5