题目描述: 小Py最近迷上了幸运数字。他所认为的幸运数字是指只由4或7组成的数字。比如44, 7774, 4都是幸运数字,而5, 17, 4437等就不是。
现在小Py有一棵树, 这棵树有n个节点, n-1条无向边且保证树连通。现在每条边都有个权值, 如果该权值为幸运数, 那么这个边就是幸运边。小Py想知道存在多少个三元组(a, b, c), 使得在树上, a到b的路径和b到c的路径上都至少存在一个幸运边。三元组的不同排序算作不同的结果。 比如(1, 2, 3), (2, 1, 3), (1, 3, 2)都是不同的三元组。
现在给你这棵树的结构L,求三元组的个数。其中L是一个三元组的列表,每个三元组(x, y, z)表示点x到y有一条边,权值为z; 节点的编号为从1开始的连续数字。
例如,L=[(1, 2, 4),(3, 1, 2),(1, 4, 7)] 则输出: 16 Note:16个三元组分别为:(1,2,4),(1,4,2),(2,1,3),(2,1,4),(2,3,1),(2,3,4),(2,4,1),(2,4,3),(3,2,4),(3,4,2),(4,1,2),(4,1,3),(4,2,1),(4,2,3),(4,3,1),(4,3,2)。
示例: 输入: L = [[1, 2, 4], [3, 1, 2], [1, 4, 7]] 输出: 16
分析: 稍等片刻。。。。
代码:
1
精彩马上继续。。。。。