题意:给出一串数字,他可能是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;
}