http://www.51nod.com/Challenge/Problem.html#!#problemId=1588
比得喜欢幸运数字。这里所说的幸运数字是由4和7组成的正整数。比如,数字47,744,4是幸运数字,而5,17,467就不是。
一天,比得遇到一棵由n个点组成的树。另外,这棵树是带权的,即每条边有一个权值(由一个正整数表示)。如果一条边的权值是一个幸运数字,那么我们就说这条边是一条幸运边。说明一下,一棵n个结点的树是由n个结点和n-1条边组的无环的无向图。
比得好奇,在树中有多少个满足以下条件的三元组tr(i,j,k)(i,j,k是三个不同的点)。
1.i到j有路径,i到k也有路径
2.每条路径中至少有一条幸运边。
数字的顺序是有意义的,举例说明,tr(1,2,3),tr(1,3,2),tr(2,1,3)是三个不同的序列。
现在要求计算在树中存在多少个这样的三元组关系。
思路:
要想不重不漏地枚举所有组合,我们枚举所有的i,对于每一个i,找到i所对应的(j,k)的组合数,加起来就行了。
令 f[u] 为在 u 的子树内,到 u 的路径中有幸运边的点有几个,令 g[u] 为在 u 的子树外,到 u 的路径中有幸运边的点有几个。
<