2-6 数列求和-加强版 (20 分)

给定某数字A(1≤A≤9)以及非负整数N(0≤N≤100000),求数列之和S=A+AA+AAA+⋯+AA⋯A(N个A)。例如A=1, N=3时,S=1+11+111=123。

输入格式:
输入数字A与非负整数N。

输出格式:
输出其N项数列之和S的值。

输入样例:
1 3

输出样例:
123

#include<bits/stdc++.h>
using namespace std;
int ans[1000005];
int main()
{
 int a,n;
 int c=0,i,x;
 cin>>a>>n;
 if(n==0) {
  cout<<0<<endl;
  return 0;
 }
 for(i=1;i<=n;i++){
  x=a*(n-i+1);
  x+=c;
  ans[i]=x%10;
  c=x/10;
 }
 if(c!=0) cout<<c;
 i--;
 for(i;i>=1;i--){
  cout<<ans[i];
 }
 cout<<endl;
 return 0; 
}