HDU 4099 Revenge of Fibonacci （数学+字典数）

传送门：http://acm.hdu.edu.cn/showproblem.php?pid=4099

 

这个题目就是一个坑或。

题意：给你不超过40的一串数字，问你这串数字是Fibonacci多少的开头几位数字，如果不存在则输出-1.

题解：明明说好的不超过40，但是在建字典数的时候不加i<41就超内存了，杭电你是想咋地，害的我比较好多人的代码，一点一点试出来的。

 

AC代码：

#include <iostream>

#include <cstdio>

#include <cstring>

#include <string>

#include <cstdlib>

#include <cmath>

#include <vector>

#include <list>

#include <deque>

#include <queue>

#include <iterator>

#include <stack>

#include <map>

#include <set>

#include <algorithm>

#include <cctype>

using namespace std;



#define si1(a) scanf("%d",&a)

#define si2(a,b) scanf("%d%d",&a,&b)

#define sd1(a) scanf("%lf",&a)

#define sd2(a,b) scanf("%lf%lf",&a,&b)

#define ss1(s)  scanf("%s",s)

#define pi1(a)    printf("%d\n",a)

#define pi2(a,b)  printf("%d %d\n",a,b)

#define mset(a,b)   memset(a,b,sizeof(a))

#define forb(i,a,b)   for(int i=a;i<b;i++)

#define ford(i,a,b)   for(int i=a;i<=b;i++)



typedef __int64 LL;

const int N=10;

const int INF=0x3f3f3f3f;

const double PI=acos(-1.0);

const double eps=1e-7;



char str[4][100];



struct Trie

{

    int v;

    Trie *next[N];

    Trie()

    {

        v=-1;

        for(int i=0;i<N;i++)

            next[i]=NULL;

    }

}*root;



void creat_trie(char s[],int x)

{

    int len=strlen(s);

    Trie *p=root;

    for(int i=0;i<len&&i<41;i++)//这个地方太肯爹了，明明说好的不超过40，不加i<41就超内存了，杭电你是想咋地

    {

        int id=s[i]-'0';

        if(p->next[id]==NULL)

            p->next[id]=new Trie();

        p=p->next[id];

        if(p->v<0)

            p->v=x;

    }

}



void add(char a[],char b[],char c[])

{

    int lena=strlen(a)-1,lenb=strlen(b)-1;

    int k=0,up=0;

    int x,y,z;

    while(lena>=0||lenb>=0)

    {

        if(lena<0) x=0;

        else x=a[lena]-'0';



        if(lenb<0)  y=0;

        else y=b[lenb]-'0';

        z=x+y+up;

        c[k++]=z%10+'0';

        up=z/10;

        lena--;

        lenb--;

    }

    if(up>0)    c[k++]=up+'0';

    c[k]=0;

    for(int i=0;i<k/2;i++)

        swap(c[i],c[k-i-1]);

//    cout<<k<<" "<<c<<endl;system("pause");

}



int find_trie(char st[])

{

    Trie *p=root;

    int len=strlen(st);

    int tmp;

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

    {

        int id=st[i]-'0';

        if(p->next[id]==NULL)

            return -1;

        else

        {

            p=p->next[id];

            tmp=p->v;

        }

    }

    return tmp;

}



void init()

{

    str[1][0]='1';  str[1][1]=0;

    creat_trie(str[1],0);

    str[2][0]='1';  str[2][1]=0;

    creat_trie(str[2],1);

    for(int i=2;i<100000;i++)//注意题目是小于，不能取等号。。

    {

        int len1=strlen(str[1]);

        int len2=strlen(str[2]);

        if(len2>60)//舍去地位

        {

            str[2][len2-1]=0;

            str[1][len1-1]=0;

        }

        add(str[1],str[2],str[3]);



        creat_trie(str[3],i);

        strcpy(str[1],str[2]);

        strcpy(str[2],str[3]);

    }

}



int main()

{

//    freopen("input.txt","r",stdin);

    root=new Trie();

    init();

    int T,ca=0;

    char st[66];

    si1(T);

    while(T--)

    {

        ss1(st);

        printf("Case #%d: %d\n",++ca,find_trie(st));

    }

    return 0 ;

}