PTA 数列求和加强版

  PTA编程题：

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

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

输入格式：

输入数字AA与非负整数NN。

输出格式：

输出其NN项数列之和SS的值。

输入样例：

1 3

输出样例：

123

这道题不能简单的用数类型来运算，因为N太大，即使long long型也存不下，所以用数组来存放和运算。

代码如下：

#include<iostream>
using namespace std;
int arr[200000] = { 0 }; //用数组来存放数，并实现进位和输出；
int main(){
int a, n,A,j=0,size=0;
cin >> a >> n;
if (n != 0)
{
for (int i = 0; i < n; i++)
{
arr[i] = (n - i)*a; //每一位先存放一个运算的结果
}
while (arr[j] != 0) //进行进位运算
{
A = arr[j] % 10;
arr[j + 1] = arr[j] / 10 + arr[j + 1];
arr[j] = A;
j++;
}
for (int i = j - 1; i >= 0; i--)
{
cout << arr[i];
}
}
else
cout << 0;
return 0;
}