hdu 4099 Revenge of Fibonacci 字典树+大数

将斐波那契的前100000个，每个的前40位都插入到字典树里（其他位数删掉），然后直接查询字典树就行。

此题坑点在于

1、字典树的深度不能太大，事实上，超过40在hdu就会MLE……

2、若大数加法时，保存的位数过少，就会导致低位误差，累积起来就可能导致前40位产生错误，解决办法是提高精度。

 

#include<iostream>

#include<cstring>

#include<string>

#include<cstdio>

using namespace std;



struct Trie

{

    struct Trie *son[10];

    int id;

}*root;

void insert(char s[],int ID)

{

    int i,j;

    struct Trie *c,*newnode;

    c=root;

    for(i=0;s[i];i++)

    {

        if(c->son[s[i]-'0'])

        {

            c=c->son[s[i]-'0'];

        }

        else

        {

            c->son[s[i]-'0']=new Trie;

            for(j=0;j<10;j++)    c->son[s[i]-'0']->son[j]=0;

            c=c->son[s[i]-'0'];

            c->id=ID;

        }

    }

}

int find(char *s)

{

    int i;

    int len;

    struct Trie *c;

    len=strlen(s);

    if(len==0)    return 0;

    c=root;

    for(i=0;i<len;i++)

    {

        if(c->son[s[i]-'0']!=0)

        {

            c=c->son[s[i]-'0'];

        }

        else

            return -1;

    }

    return c->id;

}

void init()

{

    root=new Trie;

    for(int i=0;i<10;i++)    root->son[i]=0;

}

char a[60]="1",b[60]="1",c[60];

int main()

{

    char str[50];

    int cas;

    int ca=1;

    init();

    insert("1",0);

    int la=1;

    int lb=1;

    for(int i=2;i<100000;i++)

    {

        int len=max(la,lb);

        int x=0;

        int top=0;

        for(int j=0;j<len;j++)

        {

            if(j>=la) a[j]='0';

            int s=x+a[j]+b[j]-2*'0';

            x=s/10;

            c[top++]=s%10+'0';

        }

        if(x) c[top++]=x+'0';

        c[top]='\0';

        if(top>50)              //计算数列的精度必须大一些

        {

            for(int j=0;j<49;j++) a[j]=b[j+1];a[la=49]='\0';

            for(int j=0;j<50;j++) b[j]=c[j+1];

            b[lb=50]='\0';

        }

        else

        {

            la=lb;

            strcpy(a,b);

            strcpy(b,c);

            lb=top;

        }

        int j,c;

        for(j=lb-1,c=0;j>=0&&c<40;c++,j--) str[c]=b[j];str[c]='\0';//字典树的深度绝对不能太大，超过40就要MLE！

        insert(str,i);

    }

  //  cout<<"ok"<<endl;

    cin>>cas;

    while(cas--)

    {

        scanf("%s",str);

        printf("Case #%d: %d\n",ca++,find(str));

    }

    return 0;

}