［hackerrank］Fibonacci Modified

题目链接：Fibonacci Modified

题意：就是一个斐波那契变形的数列。麻烦的是结果可能非常大（可能超过十万位？），所以必须在空间和时间上都做一些优化。

#include<stdio.h>
#include<string.h>
#define N 120005
#define mod 10000
void translate(char *s,int a[],int *a_ln)
{
    memset(a,0,N*sizeof(int));
    *a_ln=0;
    int ln=strlen(s);
    for(int i=0,base=1,temp=0;s[i]!='\0';i++)
    {
        base=1;
        for(int j=0;j<4&&s[i]!='\0';j++,i++)
        {
            a[*a_ln]=a[*a_ln]+(s[ln-i-1]-'0')*base;
            base*=10;
        }
        i--;
        (*a_ln)++;
    }
    return ;
}
void times(char *s1,char *s2,char *s)
{
    int a[N],b[N],c[N];
    int a_ln,b_ln,c_ln;
    int ans[2][N];
    int ln;
    translate(s1,a,&a_ln);
    translate(s2,b,&b_ln);
    memset(ans,0,sizeof(ans));
    ln=b_ln>a_ln?b_ln:a_ln;
    for(int i=0;i<a_ln;i++)
    {
        int k=i%2;
        int temp=0;
        int k_temp=(k+1)%2;
        memset(ans[k],0,sizeof(ans[k]));
        for(int j=0;j<b_ln;j++)
        {
            ans[k][j+i]=a[i]*b[j]+temp;
            temp=ans[k][j+i]/mod;
            ans[k][j+i]%=mod;
        }
        if(temp>0) ans[k][b_ln+i]=temp;
        if(b_ln+i>=ln) ln=b_ln+i+1;
        temp=0;
        for(int j=0;j<ln;j++)
        {
            ans[k][j]=ans[k][j]+ans[k_temp][j]+temp;
            temp=ans[k][j]/mod;
            ans[k][j]%=mod;
        }
        if(temp>0) ans[k][ln++]=temp;
    }
    int k=(a_ln+1)%2;
    c_ln=0;
    int s_ln=1;
    for(int i=0;i<ln;i++)
    {
        for(int j=0;j<4;j++)
        {
            c[c_ln++]=ans[k][i]%10;
            ans[k][i]/=10;
            if(c[c_ln-1]>0&&c_ln>s_ln) s_ln=c_ln;
        }
    }
    for(int i=0;i<s_ln;i++) s[i]=c[s_ln-i-1]+'0';
    s[s_ln]='\0';
    return ;
}
void add(char *s1,char*s2,char *s)
{
    int ln1=strlen(s1);
    int ln2=strlen(s2);
    int ln=ln1>ln2?ln1:ln2;
    int temp=0;
    for(int i=0;i<ln;i++)
    {
        int s1_temp,s2_temp,s_temp;
        s1_temp=i>=ln1?0:s1[ln1-i-1]-'0';
        s2_temp=i>=ln2?0:s2[ln2-i-1]-'0';
        s_temp=s1_temp+s2_temp+temp;
        temp=s_temp/10;
        s_temp%=10;
        s[i]=s_temp+'0';
    }
    if(temp>0) s[ln++]=temp+'0';
    s[ln]='\0';
    for(int i=0;i<ln/2;i++)
    {
        char c_temp;
        c_temp=s[i];
        s[i]=s[ln-i-1];
        s[ln-i-1]=c_temp;
    }
    return ;
}
int main()
{
    char s[21][N];
    int n;
    scanf("%s%s%d",s[1],s[2],&n);
    for(int i=3;i<=n;i++)
    {
        char s_temp[N];
        times(s[i-1],s[i-1],s_temp);
        add(s_temp,s[i-2],s[i]);
    }
    puts(s[n]);
    return 0;
}