一道字典树的应用。
这一题做起来,想要AC还是需要扎实的功底啊。我做起来就感觉,到处都不踏实,那就好好捋一捋。
首先,斐波那契数的计算,这个就没弄好,先贴公式f(0) = 0, f(1) = 1, f(n) = f(n - 1) + f(n - 2)。那么为什么这一题输入1,输出却是0。那么肯定是f(0) = 1, f(1) = 1。(纯属猜测)
计算数列的时候,也感觉不好下手,如果是用一个一维数列的话,完全没有头绪。参考大神们,发现他们用了3个数列,分别存储f(n), f(n - 1), f(n - 2)。那么进行下一步的时候,需要f(n + 1) , f(n - 1), f(n),后两项就可以由前面转移啦。注意精度和memory,特别是字典树千万不要存40位后太多。我刚存40位,交G++就已经mle了。又是只能交C++,至今不懂。
/**
TASK: crypt1
LANG: C++
ID: DickensTone
**/
#include<cstdio>
#include<cstring>
#include<fstream>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 65;
int f1[maxn + 5], f2[maxn + 5], f3[maxn + 5];
char str[maxn + 5];
struct node
{
int id;
struct node *nex[10];
node()
{
id = -1;
for(int i = 0; i < 10; i++)
nex[i] = NULL;
}
};
node *root;
void insert_tree(char *p, int ans)
{
node *q = root;
for(int i = 0; i < strlen(str); i++)
{
//printf("yes\n");
int ID = p[i] - '0';
if(q->nex[ID] == NULL)
{
q -> nex[ID] = new node();
}
q = q->nex[ID];
if(q->id < 0) q->id = ans;
//printf("yes\n");
}
}
int get(char *p)
{
node *q= root;
for(int i = 0; i < strlen(str); i++)
{
int id = p[i] - '0';
if(q->nex[id] == NULL) return -1;
else q = q->nex[id];
}
return q->id;
}
void init()
{
memset(f1, 0, sizeof(f1));
memset(f2, 0, sizeof(f2));
memset(f3, 0, sizeof(f3));
memset(str, 0, sizeof(str));
str[0] = '1';
str[1] = 0;
insert_tree(str, 0);
f1[0] = f2[0] = 1;
for(int p = 2; p < 100000 ; p++)
{
int r = 0;
for(int i = 0; i < maxn; i++)
{
f3[i] = f1[i] + f2[i] + r;
r = f3[i] / 10;
f3[i] = f3[i] % 10;
}
int last = 0;
for(int i = maxn; i >= 0; i--)
if(f3[i])
{
last = i;
break;
}
int k = 0;
//printf("yes\n");
for(int i = 0; i <= last && k <= 40; i++) str[k++] = f3[last - i] + '0';
str[k] = 0;
insert_tree(str, p);
int s = last > 55 ? last - 55 : 0;
for(int i = 0; i + s <= last + 5; i++)
f1[i] = f2[s + i];
for(int i = 0; i + s <= last + 5; i++)
f2[i] = f3[s + i];
}
}
int main()
{
//freopen("crypt1.in", "r", stdin);
//freopen("crypt1.out", "w", stdout);
root = new node();
init();
int T, kase = 1;
scanf("%d", &T);
while(T--)
{
scanf("%s", str);
printf("Case #%d: %d\n", kase++, get(str));
}
return 0;
}