PTA 7-38 数列求和-加强版

7-38 数列求和-加强版（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 <stdio.h>
#include <stdlib.h>
 
int main()
{
    int a = 0;
    int n = 0;
    scanf("%d %d", &a, &n);
    int flag = 0;
    int tmp = 0;
    int* sum = (int*)malloc(sizeof(int) * (n + 1)); // 给sum开辟空间
    if (a == 0 || n == 0)
    {
        printf("0");
    }
    else
    {
        for (int i = 0; i < n; i++)
        {
            if (i == 0)
            {
                sum[i] = a * (n - i) % 10;
            }
            else
            {
                sum[i] = (a * (n - i) % 10 + tmp)%10;
            }
            tmp = (a * (n - i)  + tmp) / 10;
            if (i == n - 1 && tmp != 0)
            {
                sum[n] = 1;
                flag = 1;
            }
        }
    }

    if (flag == 1)
    {
        for (int i = n; i >= 0; i--)
        {
            printf("%d", sum[i]);
        }
    }
    else
    {
        for (int i = n-1; i >= 0; i--)
        {
            printf("%d", sum[i]);
        }
    }

    return 0;
}