【SCOI2003】【BZOJ1089】严格n元树

Description

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

给出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
HINT

Source

用排列组合推一推发现递推式

f[i]=f[i-1]^n+1
ans=f[d]-f[d-1]

高精度所以Python大法
答案太长了光输出就T了
然而数据傻逼

#本来应该A的Python3代码...但是BZOJ上WA了
n,d=map(int, input().split(' '))
if d==0:
        print(1)
else:
        f=[1]
        for i in range(0,d+1):
                f.append(f[i]**n+1)
        print(f[d]-f[d-1])

但是换成Python2.7写法就A了

n,d=map(int, raw_input().split())
if d==0:
    print 1
else:
    f=[1]
    for i in range(0, d+1):
        f.append(f[i]**n+1)
    print f[d]-f[d-1]