自己很久都没写过字典树了,直接转了别人的代码,嘿嘿。。
题意:输入一个数n,长度至多40,输入最小序号i,使得f[i]的前缀是n,若i大于1e5则输出-1。f[i]表示第i个斐波那契数列,f[0]=f[1]=1,f[i]=f[i-1]+f[i-2];
题解:可以说是一道字典树模板题。就是用大数加法预处理小于100000的斐波那契数列,将小于40的前缀保存在字典树中。然后每次询问只要在字典树种找就行了。字典树:就是单词查找树,我们查单词的时候是先查单词第一个字母,然后第二个。。直到找到为止;字典树就是以这种原理保存数据。
注意:
1.由于精度问题,所以前缀取60位左右。。
耗时:640MS/5000MS
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=10;
struct node{
int w;
struct node *next[maxn];
node():w(-1){memset(next,NULL,sizeof(next));}
}root;
char d[100];
void add(char *a,char *b,char *c)
{
int i,j,k,aa,bb,t=0,p=0;
aa=strlen(a)-1;
bb=strlen(b)-1;
while(aa>=0||bb>=0)
{
if(aa<0)i=0;
else i=a[aa]-'0';
if(bb<0)j=0;
else j=b[bb]-'0';
k=i+j+t;
d[p++]=k%10+'0';
t=k/10;
aa--;
bb--;
}
while(t)
{
d[p++]=t%10+'0';
t=t/10;
}
for(i=0;i<p;i++)c[i]=d[p-i-1];
c[p]='\0';
}
void insert(char *a,int x)
{
node *p=&root;
int i,t=0;
for(i=0;a[i]!='\0'&&i<=40;i++)
{
if(p->next[a[i]-'0']==NULL){p->next[a[i]-'0']=new node;}
p=p->next[a[i]-'0'];
if(p->w==-1)p->w=x;
}
}
int find(char *a)
{
node *p=&root;
int i;
for(i=0;a[i]!='\0';i++)
{
if(p->next[a[i]-'0']!=NULL)p=p->next[a[i]-'0'];
else return -1;
}
return p->w;
}
void init()
{
int i,j,k;
char a[100],b[100],c[100],aa,bb;
a[0]=b[0]='1';
a[1]=b[1]='\0';
insert(a,0);
for(i=2;i<100000;i++)
{
aa=strlen(a);
bb=strlen(b);
if(bb>60)a[aa-1]=b[bb-1]='\0';
add(a,b,c);
insert(c,i);
strcpy(a,b);
strcpy(b,c);
}
}
int main()
{
int T,tt=0;
char a[100];
init();
scanf("%d",&T);
while(T--)
{
scanf("%s",a);
printf("Case #%d: %d\n",++tt,find(a));
}
return 0;
}