P4295 [SCOI2003]严格N元树 DP
思路:DP
提交:\(5\)次
错因:2次高精写错(我太菜了),2次写错特判
题解:
设\(f[i]\)表示深度\(\leq i\)的严格\(n\)元树的数目,有
\[f[i]=pow(f[i-1],n)+1
\]
即一个点,对于每一个孩子深度都可以是\(1\)到\(i-1\)的严格\(n\)元树,或是仅仅一个点(作为根)。
所以最后的答案是\(f[i]-f[i-1]\)
需要高精。
#include<cstdio>
#include<iostream>
#include<cstring>
#define R register int
using namespace std;
namespace Luitaryi {
template<class I> inline I g(I& x) { x=0;
register I f=1; register char ch; while(!isdigit(ch=getchar())) f=ch=='-'?-1:f;
do x=x*10+(ch^48); while(isdigit(ch=getchar())); return x*=f;
} const int N=10010;
int n,d,a[N],b[N],f[N],mem[N],sza,szb;
inline void print(int* f,int len) {
for(R i=len;i;--i) cout<<f[i]; cout<<endl;
}
inline void mul(int a[N],int b[N],int& sza,int szb) {
R tmp[N]; memset(tmp,0,sizeof(tmp));
for(R i=1;i<=sza;++i) for(R j=1;j<=szb;++j)
tmp[i+j-1]+=a[i]*b[j];
sza+=szb-1; for(R i=1;i<=sza;++i) tmp[i+1]+=tmp[i]/10,tmp[i]%=10;
while(tmp[sza+1]) ++sza,tmp[sza+1]+=tmp[sza]/10,tmp[sza]%=10;
memcpy(a,tmp,sizeof(int)*(sza+3));
}
inline void main() {
g(n),g(d); if(d==0) return (void) puts("1"); f[1]=1,sza=1;
for(R i=1;i<=d;++i) { memcpy(mem,f,sizeof(int)*(sza+3)),szb=sza;
for(R j=2;j<=n;++j) mul(f,mem,sza,szb); ++f[1];
for(R j=1;j<=sza;++j) f[j+1]+=f[j]/10,f[j]%=10;
while(f[sza+1]) ++sza,f[sza+1]+=f[sza]/10,f[sza]%=10;
} for(R i=1;i<=sza;++i) f[i]-=mem[i];
for(R i=1;i<=sza;++i) if(f[i]<0) f[i]+=10,--f[i+1];
while(!f[sza]) --sza; while(sza) printf("%d",f[sza]),--sza;
}
} signed main() {Luitaryi::main(); return 0;}
2019.08.16
84