PTA 数列求和加强版

  PTA编程题：

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

这道题不能简单的用数类型来运算，因为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;
}