probelm D 大数+组合dp

#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
#define maxn 100
struct Bignum
{
  int num[maxn];
  int len;
};
Bignum f[33][33];
Bignum Add(Bignum &a,Bignum &b)
{
    Bignum c;
    int len=(a.len>b.len)?a.len:b.len;
    memset(c.num,0,sizeof(c.num));
    for(int i=0;i<len;i++)
    {
     c.num[i]+=(a.num[i]+b.num[i]);
     if(c.num[i]>=10)
     {
      c.num[i+1]++;
      c.num[i]-=10;
     }
    }
    if(c.num[len])len++;
    c.len=len;
    return c;
}
Bignum Mult(Bignum &a,int b)
{
    Bignum c;
    int len=a.len;
    memset(c.num,0,sizeof(c.num));
    for(int i=0;i<len;i++)
    {
        c.num[i]+=(a.num[i]*b);
        if(c.num[i]>=10)
        {
            c.num[i+1]=c.num[i]/10;
            c.num[i]%=10;
        }
    }
    while(c.num[len]>0)
    {
            c.num[len+1]=c.num[len]/10;
            c.num[len++]%=10;
    }
    c.len=len;
    return c;
}
void print(Bignum &a)
{
    for(int i=a.len-1;i>=0;i--)
        printf("%d",a.num[i]);
    putchar(10);
}
int main()
{
   int len;
    int P,L,K;
    Bignum  ans;
    while(~scanf("%d%d%d",&P,&L,&K))
    {
    for(int i=0;i<33;i++)
        for(int j=0;j<33;j++)
    {
         memset(f[i][j].num,0,sizeof (f[i][j].num));
         f[i][j].len=0;
    }
    f[1][1].len=1;
    f[1][1].num[0]=1;
    int x=P-1;
    int len=0;
    while(x)
    {
        f[1][0].num[len++]=x%10;
        x/=10;
    }
    f[1][0].len=len;
    for(int i=2;i<=L;i++)
    {
         ans.len=0;
         memset(ans.num,0,sizeof(ans.num));
        for(int j=K-1;j>=1;j--)
        {
            f[i][j]=f[i-1][j-1];
            ans=Add(ans,f[i][j]);
        }
        Bignum k=Add(ans,f[i-1][K-1]);
        f[i][0]=Mult(k,P-1);
    }
    ans.len=1;
    memset(ans.num,0,sizeof(ans.num));
    ans.num[0]=0;
    for(int i=0;i<=K-1;i++)
    {
       ans=Add(ans,f[L][i]);
    }
    for(int i=ans.len-1;i>=0;i--)
     printf("%d",ans.num[i]);
    putchar(10);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/kala0/article/details/72628603