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>
using namespace std;
int main(void){ 
    int A, N,flag,temp;
    long j=0;
    int *num;
    cin >> A >> N;
    if (N == 0){
        cout << N;
    }
    else{
        num = new int[N];
        flag = 0;
        for (int i = N; i >= 1; i--){
            temp = A*i + flag;
            flag = temp / 10;
            num[j] = temp % 10;
            j++;
        }
        if (flag > 0){
            num[j] = flag;
        }
        else{
            j--;
        }
        for (int i = j; i >= 0; i--){
            cout << num[i];
        }
        delete[] num;
    }
    return 0; 
}