题目1051：数字阶梯求和 2010年哈尔滨工业大学计算机研究生机试真题

题目1051：数字阶梯求和

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：4894

解决：1664

题目描述：

给定a和n，计算a+aa+aaa+a...a(n个a)的和。

输入：

测试数据有多组，输入a，n（1<=a<=9,1<=n<=100）。

输出：

对于每组输入,请输出结果。

样例输入：

1 10

样例输出：

1234567900

来源：

2010年哈尔滨工业大学计算机研究生机试真题

#include <stdio.h>
#include <iostream>
typedef struct BigInt BigInt;
struct BigInt{
    int digit[1000];
    int size;
    void init(){
        for(int i=0;i<1000;i++){
            digit[i]=0;
        }
        size=0;
    }
    void set(int a,int n){
        init();
        int j=0,t=0,c=1;
        for(int i=n-1;i>=0;i--){
            t+=a*c;
            j++;c*=10;
            if(j==4||i==0){
                digit[size]=t;
                size++;j=0;t=0;c=1;
            }
        }
    }
    void output()
    {
        for(int i=size-1;i>=0;i--){
            if(i!=size-1)
                printf("%04d",digit[i]);
            else
                printf("%d",digit[i]);
        }
        printf("\n");
    }
    BigInt operator +(const BigInt &A)const{
        BigInt res;res.init();
        int carry=0;
        for(int i=0;i<A.size||i<size;i++){
            int tmp=A.digit[i]+digit[i]+carry;
            carry=tmp/10000;
            tmp%=10000;
            res.digit[res.size]=tmp;
            res.size++;
        }
        if(carry!=0){
            res.digit[res.size++]=carry;
        }
        return res;
    }
};
int main(){
    int a=0,n=0;
    while (~scanf("%d %d",&a,&n)){
        BigInt res,x;
        res.init();
        res.set(0,1);
        for(int i=1;i<=n;i++)
        {
            x.init();
            x.set(a,i);
            res=res+x;
        }
        res.output();
    }
    return 0;
}
/**************************************************************
    Problem: 1051
    User: 爱理momoko
    Language: C++
    Result: Accepted
    Time:0 ms
    Memory:1520 kb
****************************************************************/