【BZOJ 1089】 [SCOI2003]严格n元树

1089: [SCOI2003]严格n元树

Time Limit: 1 Sec Memory Limit: 162 MB

Submit: 872 Solved: 445

[Submit][Status]

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

dp+高精度。

f[i]表示深度<=i的n元树有几种。

f[0]=1

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

这里的+1是加上深度为0的情况。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;
struct data
{
int l,v[1005];
}f[20];
int n,d;
data operator *(data a,data b)
{
data ans;
for (int i=1;i<=1000;i++)
ans.v[i]=0;
ans.l=0;
for (int i=1;i<=1000;i++)
ans.v[i]=0;
for (int i=1000;i>=1001-b.l;i--)
for (int j=1000;j>=1001-a.l;j--)
ans.v[i-(1000-j)]+=(b.v[i]*a.v[j]);
int now=1001-(a.l+b.l-1);
for (int i=1000;i>now;i--)
{
if (ans.v[i]<10) continue;
ans.v[i-1]+=(ans.v[i]/10);
ans.v[i]%=10;
}
ans.l=a.l+b.l-1;
while (ans.v[now]>=10)
{
ans.l++;
ans.v[now-1]+=(ans.v[now]/10);
ans.v[now]%=10;
now--;
}
return ans;
}
data operator ^(data a,int x)
{
data ans;
for (int i=1;i<=1000;i++)
ans.v[i]=0;
ans.l=1,ans.v[1000]=1;
while (x)
{
if (x&1) ans=ans*a;
a=a*a;
x>>=1;
}
for (int i=1;i<=1000;i++)
if (ans.v[i])
{
ans.l=1001-i;
break;
}
return ans;
}
data operator +(data a,int x)
{
a.v[1000]+=x;
int now=1000;
while (a.v[now]>=10)
{
a.v[now-1]+=(a.v[now]/10);
a.v[now]%=10;
now--;
}
for (int i=1;i<=1000;i++)
if (a.v[i])
{
a.l=1001-i;
break;
}
return a;
}
data operator -(data a,data b)
{
for (int i=1000;i>=1001-a.l;i--)
{
a.v[i]-=b.v[i];
if (a.v[i]<0)
a.v[i]+=10,a.v[i-1]--;
}
while (!a.v[1001-a.l])
a.l--;
return a;
}
void Print(data a)
{
for (int i=1001-a.l;i<=1000;i++)
printf("%d",a.v[i]);
cout<<endl;
}
int main()
{
scanf("%d%d",&n,&d);
f[0].l=1,f[0].v[1000]=1;
for (int i=1;i<=d;i++)
f[i]=(f[i-1]^n)+1;
Print(f[d]-f[d-1]);
return 0;
}

感悟：

1.WA多次，是对于数字长度的处理出错了

2.一开始对于这道题想的是f[i]表示深度为i的n元树个数，发现很难转移；设成<=i的就方便多了