HDU4099 Revenge of Fibonacci

来源:互联网 发布:王充 论衡 知乎 编辑:程序博客网 时间:2024/06/01 11:30

题意:给出一串数字,他可能是Fibonacci数的开头,也有可能不是,是的话输出他的编号(具有这个前缀最小的编号),比如编号0是1,编号1也是1,编号2就是2,编号3就是3.。编号4是5.。不是输出-1.。。

trie+高精度加法。。可以从样例看出Fibonacci数的值一定很大(样例都快到第5W个Fibonacci数了),但是题目最多输入的前缀长度小于40所以,对于字典树来说后面的很多位数都能够省掉(60,为什么是60我也不知道),然后把Fibonacci数建立字典树就行了,因为输入的前缀小于40,所以建立字典树最多的长度只要<40就好,字典树上每条边都直接用编号覆盖,覆盖过了不覆盖,因为肯定是小的先覆盖。。

还有一个我WA了很多次的地方,Fibonacci数是插入到<10W,我写成<=10W就WA了n次。。。简直了。。。

#include<cstdio>#include<cstring>#include<algorithm>#include<cstdlib>using namespace std;struct trie{    trie *next[10];    int val;    trie():val(-1){};}root;void add(char a[],char b[],char back[]){    int i,j,k,up,x,y,z,l;    char *c;    if (strlen(a)>strlen(b)) l=strlen(a)+2; else l=strlen(b)+2;          c=(char *) malloc(l*sizeof(char));    i=strlen(a)-1;    j=strlen(b)-1;    k=0;up=0;    while(i>=0||j>=0)    {        if(i<0) x='0'; else x=a[i];        if(j<0) y='0'; else y=b[j];        z=x-'0'+y-'0';        if(up) z+=1;        if(z>9) {up=1;z%=10;} else up=0;        c[k++]=z+'0';        i--;j--;    }    if(up) c[k++]='1';    i=0;    c[k]='\0';    for(k-=1;k>=0;k--)        back[i++]=c[k];    back[i]='\0';}void insert(char *str,int val){    trie *p=&root,*q;    int len=strlen(str);    for(int i=0;i<len&&i<40;i++)    {        int id=str[i]-'0';        if(p->next[id]==NULL)        {            q=(trie *)malloc(sizeof(root));            q->val=-1;            for(int j=0;j<10;j++)                q->next[j]=NULL;            p->next[id]=q;        }        p=p->next[id];        if(p->val<0)            p->val=val;    }}int findx(char *str){    trie *p=&root;    int len=strlen(str);    int cnt;    for(int i=0;i<len;i++)    {        int id=str[i]-'0';        if(p->next[id]!=NULL)        {            p=p->next[id];            cnt=p->val;        }        else            return -1;    }    return cnt;}char num[3][62];int main(){    int t,i,flag=1;    char n[60];    num[0][0]='1';    num[0][1]='\0';    insert(num[0],0);    num[1][0]='1';    num[1][1]='\0';    insert(num[1],1);    for(i=2;i<100000;i++)    {        int len1=strlen(num[0]);        int len2=strlen(num[1]);        if(len2>60)        {            num[0][len1-1]='\0';            num[1][len2-1]='\0';        }        add(num[0],num[1],num[2]);        insert(num[2],i);        strcpy(num[0],num[1]);        strcpy(num[1],num[2]);    }    scanf("%d",&t);    while(t--)    {        scanf("%s",n);        printf("Case #%d: %d\n",flag++,findx(n));    }    return 0;}


0 0
原创粉丝点击