BZOJ 1089 严格n元树 (递推+高精度)

题解:用a[i]表<=i时有几种树满足度数要求,那么这样就可以递归了,a[i]=a[i-1]^n+1。n个节点每个有a[i-1]种情况,那么将其相乘,最后加上1,因为深度为0也算一种。那么答案就是a[n]-a[n-1]。然后就是高精度的问题了,发现很久没有现码高精度没手感了,连高进度加法进位都出了些问题,需要特别注意。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct data{int len,a[2002];}a[35],c,p,t;
int n,d;
data mul(data a,data b){
    memset(c.a,0,sizeof c.a);
    for(int i=1;i<=a.len;i++)
    for(int j=1;j<=b.len;j++){
        c.a[i+j-1]+=a.a[i]*b.a[j];
        c.a[i+j]+=c.a[i+j-1]/10000;
        c.a[i+j-1]%=10000;
    }c.len=2000;
    while(c.len&&!c.a[c.len])c.len--;
    return c;
}
data sum(data a,data b){
    memset(c.a,0,sizeof c.a); 
    c.len=max(a.len,b.len);
    for(int i=1;i<=c.len;i++){
        c.a[i]+=a.a[i]+b.a[i];
        c.a[i+1]+=c.a[i]/10000;
        c.a[i]%=10000;
    }c.len=2000;
    while(c.len&&!c.a[c.len])c.len--;
    return c;
}
data sub(data a,data b){
    memset(c.a,0,sizeof c.a);
    c.len=a.len;
    for(int i=1;i<=a.len;i++){
        c.a[i]=a.a[i]-b.a[i];
        if(c.a[i]<0)c.a[i]+=10000,a.a[i+1]--;
    }while(c.len&&!c.a[c.len])c.len--;
    return c;
}
data power(data a,int b){
    memset(p.a,0,sizeof p.a); p.len=1; p.a[1]=1;
    while(b){
        if(b&1)p=mul(p,a);
        b>>=1; a=mul(a,a); 
    }return p;
}
data op(data a,int b){
    t.len=1; t.a[1]=1;
    return sum(power(a,b),t);
}
int main(){
    scanf("%d%d",&n,&d);
    if(!d)return puts("1"),0;
    a[0].len=1; a[0].a[1]=1;
    for(int i=1;i<=d;i++)a[i]=op(a[i-1],n);
    a[d]=sub(a[d],a[d-1]);
    printf("%d",a[d].a[a[d].len]);
    for(int i=a[d].len-1;i;i--)printf("%04d",a[d].a[i]);
    return 0;
}
posted @ 2014-07-21 10:41  forever97  阅读(365)  评论(0编辑  收藏  举报