Ural(Timus) 1012. K-based Numbers. Version 2

DP+高精度  （和1009是一样的题目，不过数字的位数达到了180，所以要用高精度）

同样是记忆化搜索实现，不过加入了高精度，一些细节地方就注意一下。我的代码写的不好，有点长有点乱…………

 

#include <cstdio>
#include <cstring>
#define LEN 210  //高精度数组的大小
#define MAX 210  //位数
struct num
{
    int a[LEN],len; //a数组保存高精度
}dp[MAX][15];  //最多180位，最高进制为10
int N,K;

void init()
{
    for(int i=1; i<N; i++)
        for(int j=0; j<K; j++)
        {
            memset(dp[i][j].a,0,sizeof(dp[i][j].a));
            dp[i][j].len=-1;
        }
    for(int i=0; i<K; i++)
    {
        dp[N][i].len=1;
        memset(dp[N][i].a,0,sizeof(dp[N][i].a));
        dp[N][i].a[0]=1;
    }
    return ;
}

void add(struct num p ,struct num q , struct num *ans)
{
    struct num tmp;
    if(q.len>p.len) { tmp=q; q=p; p=tmp; }
    int t,c=0;
    for(int i=0; i<p.len; i++)
    {
        t=p.a[i]+q.a[i]+c;
        (*ans).a[i]=t%10;
        c=t/10;
    }
    (*ans).len=p.len;
    if(c>0)
    {
        (*ans).a[p.len]=1;
        (*ans).len++;
    }
    return ;
}

void dfs(int i , int j)
{
    if(dp[i][j].len!=-1)
        return ;
    dp[i][j].len=0;

    for(int k=1; k<K; k++)
    {
        dfs(i+1,k);
        //dp[i][j]+=dp[i+1][k]
        add(dp[i][j] , dp[i+1][k] , &dp[i][j]);
    }
    if(j)
    {
        dfs(i+1,0);
        //dp[i][j]+=dp[i+1][0];
        add(dp[i][j] , dp[i+1][0] , &dp[i][j]);
    }
    return ;
}

void solve()
{
    struct num ans;
    memset(ans.a,0,sizeof(ans.a)); ans.len=0;
    
    for(int i=1; i<K; i++)
    {
        dp[1][i].len=0;
        for(int j=0; j<K; j++)
        {
            dfs(2,j);
            //dp[1][i]+=dp[2][j]
            add(dp[1][i] , dp[2][j] , &dp[1][i]);
        }
        add(ans , dp[1][i] , &ans);
    }

    for(int i=ans.len-1; i>=0; i--)
        printf("%d",ans.a[i]);
    printf("\n");
}

int main()
{
    while(scanf("%d%d",&N,&K)!=EOF)
    {
        init();
        solve();
    }
    return 0;
}