bzoj 1089 [SCOI2003]严格n元树（DP+高精度）

 

1089: [SCOI2003]严格n元树

Time Limit: 1 Sec  Memory Limit: 162 MB
Submit: 1250  Solved: 621
[Submit][Status][Discuss]

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

 

【思路】

       DP+高精度。

       设f[i]表示高i的严格n元数的数目，并设S[i]表示f[i]的前缀和。一颗高i的严格n元树有一个根节点以及n个高不超过i-1的子树构成，每个子树方案为S[n-1]，则有转移式：

       S[i]=(S[i-1]^n)+1

       1表示只有一个根的情况。

       高精度照着别人的写的，重载运算符，用起来比较方便。

 

【代码】

#include<cstdio>
#include<cstring>
#include<iostream> 
using namespace std;

const int maxn = 5000+10;
const int rad = 1000;

struct Bign {  int N[maxn],len;  };
void print(Bign a) {
    printf("%d",a.N[a.len-1]);
    for(int i=a.len-2;i>=0;i--)
        printf("%03d",a.N[i]);            //补全位数 
    putchar('\n');
}
Bign operator *(Bign A,Bign B) {
    Bign C;
    int lena=A.len,lenb=B.len;
    for(int i=0;i<lena+lenb;i++) C.N[i]=0;
    for(int i=0;i<lena;i++)
        for(int j=0;j<lenb;j++)
            C.N[i+j] += A.N[i]*B.N[j];
    C.len=A.len+B.len;
    for(int i=0;i<C.len;i++) 
        if(C.N[i]>=rad) {
            if(i==C.len-1) 
                C.len++ , C.N[i+1]=C.N[i]/rad;
            else C.N[i+1]+=C.N[i]/rad;
            C.N[i]%=rad;
        }
    while(C.len && !C.N[C.len-1]) C.len--;
    return C;
}
Bign operator ^(Bign A,int p) {            //快速幂 
    Bign C;
    C.len=1; C.N[0]=1;
    while(p) {
        if(p&1) C=C*A; A=A*A;  p>>=1;
    }
    return C;
}
Bign operator +(Bign A,int x) {
    A.N[0]+=x;
    int now=0;
    while(A.N[now]>=rad) {
        A.len=max(A.len,now+1);
        A.N[now+1]+=A.N[now]/rad;
        A.N[now]%=rad;
        now++;
        A.len=max(A.len,now);
    }
    return A;
}
Bign operator-(Bign A,Bign B) {
    for(int i=0;i<A.len;i++) {
        A.N[i]-=B.N[i];
        if(A.N[i]<0)
            A.N[i]+=rad , A.N[i+1]--;
    }
    while(A.len && !A.N[A.len-1]) A.len--;
    return A;
}

int n,d;
Bign S[50];

int main() {
    scanf("%d%d",&n,&d);
    if(!d) { puts("1"); return 0; }
    S[0].len=1; S[0].N[0]=1;
    for(int i=1;i<=d;i++)
        S[i]=(S[i-1]^n)+1;
    print(S[d]-S[d-1]);
    return 0;
}