[BZOJ1089][SCOI2003]严格n元树

1089: [SCOI2003]严格n元树

Time Limit: 1 Sec  Memory Limit: 162 MB Submit: 1925  Solved: 966 [Submit][Status][Discuss]

Description

  如果一棵树的所有非叶节点都恰好有n个儿子,那么我们称它为严格n元树。如果该树中最底层的节点深度为d (根的深度为0),那么我们称它为一棵深度为d的严格n元树。例如,深度为2的严格2元树有三个,如下图:

  给出n, d,编程数出深度为d的n元树数目。

Input

  仅包含两个整数n, d( 0   <   n   <   =   32,   0  < =   d  < = 16)

Output

  仅包含一个数,即深度为d的n元树的数目。

Sample Input

【样例输入1】
2 2

【样例输入2】
2 3

【样例输入3】
3 5

Sample Output

【样例输出1】
3

【样例输出2】
21

【样例输出2】
58871587162270592645034001
 
 设$dp[i]$表示深度不超过$i$的严格$n$元树
那么$dp[i]=dp[i-1]^{n}+1$
$ans=dp[d]-dp[d-1]$
Python不用写高精度,美滋滋
import sys
n, d = map(int, sys.stdin.readline().split())
if d == 0:
    print(1)
else:
    f = []
    f.append(1)
    for i in range(1, d + 1):
        f.append(f[i - 1] ** n + 1)
    print f[d] - f[d - 1]

 

 
posted @ 2017-10-09 19:24  jzyy  阅读(115)  评论(0编辑  收藏  举报