hdu 4099 Revenge of Fibonacci 大数加法+字典树

最新推荐文章于 2020-08-26 17:18:11 发布

转载最新推荐文章于 2020-08-26 17:18:11 发布 · 1.2k 阅读

字典树专栏收录该内容

6 篇文章

订阅专栏

本文详细介绍了如何使用字典树解决斐波那契数列的前缀匹配问题，包括预处理、字典树结构设计以及算法实现步骤。重点突出算法效率与精度考虑。

自己很久都没写过字典树了，直接转了别人的代码，嘿嘿。。

题意：输入一个数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;
}