PTA 5-38 数列求和-加强版

5-38 数列求和-加强版   (20分)

给定某数字$A$（$1\le A\le 9$）以及非负整数$N$（$0\le N\le 100000$），求数列之和$S=A+AA+AAA+\cdots +AA\cdots A$（$N$个$A$）。例如$A=1$, $N=3$时，$S=1+11+111=123$。

输入格式：

输入数字$A$与非负整数$N$。

输出格式：

输出其$N$项数列之和$S$的值。

输入样例：

1 3

输出样例：

123




#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=100005;
int ary[N];
int main()
{
    int a,n,i,j,k;
    cin>>a>>n;
    if(n==0)
    {
        cout<<0;
        return 0;
    }
    for(i=1;i<=n;++i)
    {
        ary[i]=i*a;
    }
    for(i=n;i>=1;--i)
    {
        ary[i-6]+=ary[i]/1000000;
        ary[i-5]+=(ary[i]/100000)%10;
        ary[i-4]+=(ary[i]/10000)%10;
        ary[i-3]+=(ary[i]/1000)%10;
        ary[i-2]+=(ary[i]/100)%10;
        ary[i-1]+=(ary[i]/10)%10;
        ary[i]%=10;
    }
    if(ary[0])
        cout<<ary[0];
    for(i=1;i<=n;++i)
    {
        cout<<ary[i];
    }
}