PTA基础编程题目集 7-38 数列求和-加强版

给定某数字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 <iostream>
#include <vector>
using namespace std;

int main()
{
	int a, n;
	while (true)
	{
		cin >> a >> n;
		if (a >= 1 && a <= 9 && n >= 0 && n <= 100000)
			break;
	}
	vector<int> s;
	int temp;
	int x = 0, y = 0;
	if (n > 0)
	{
		for (int i = 0; i < n; i++)
		{
			temp = a * (n - i) + x;
			x = temp / 10; //进位
			y = temp % 10; //余数
			s.push_back(y);
		}
		if (x > 0)
			cout << x;
		for (int j = s.size() - 1; j >= 0; j--)
			cout << s[j];
	}
	else
		cout << 0;

	return 0;
}