数列求和-加强版 数组

问题描述
给定某数字A（1≤A≤9）以及非负整数（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>
using namespace std;
#include <vector>
#include <malloc.h> 
#include <cstring>
//先将各项表示出来
#define Max 1000000
int main()
{
    int n;
    int A,N;
    cin>>A>>N;
    int k = N+1;//位数
   // int m = 1;
    int* nums = (int*)malloc(sizeof(int)*Max);
    memset(nums,0,sizeof(int)*Max);
    for(int i=1;i<=N;i++)
    {
        for(int j=1;j<=i;j++)
        {
            nums[j] += A;
            while(nums[j]>=10)
           {
            n = nums[j]/10;//标志进位
            nums[j] = nums[j]%10;
            nums[j+1] += n;
          }
        }
    }
    while(nums[k]>=10)
    {
        n = nums[k]/10;//标志进位
            nums[k] = nums[k]%10;
            nums[k+1] = n;
        k++;
    }
    if(nums[k]==0&&k!=1)
    {
        k = k-1;
    }
    for(int i=k;i>=1;i--)
    {
        cout<<nums[i];
    }
} 

问题分析：

1. 数字很大，所以一定用数组存储
2. 此代码在修改前未考虑N为0的情况，导致测试点2未通过
3. 这个算法的时间复杂度是O(n^2)当N、A取最大，测试点三未通过

将加法改为乘法后测试点1一直不通过，不清楚原因

#include <iostream>
using namespace std;
#include <vector>
#include <malloc.h> 
#include <cstring>
//先将各项表示出来
#define Max 1000000
int main()
{//相加时间复杂度太大，考虑相乘 
	int n = 0;//进位数 
    int A,N;
    cin>>A>>N;
    int k = N-1;//位数
    //int m = N;//标记相乘的个数 
   // int m = 1;
    int* nums = (int*)malloc(sizeof(int)*Max);
    memset(nums,0,sizeof(int)*Max);
    for(int i=0;i<N;i++)
	{
		nums[i] = (N-i)*A+n;
		if(nums[i]>=10)
		{
			n = nums[i]/10;
			nums[i] = nums[i]%10;
		}
	}
//	n = 0;
	if(k==-1)
	{
		cout<<"0";
	} 
	while(n!=0)
	{
		nums[k+1] += n; 
	//	nums[k] = nums[k]%10;
        n = nums[k+1]/10; 
		k++;
	}
	for(int i=k;i>=0;i--)
	{
		cout<<nums[i];
	}
} 

总结

1. 相乘更快，寻找规律
2. 这里位数最多为N+1所以不用设置那么多空间

n久以后写的修改版
AC代码：

#include <bits/stdc++.h>
using namespace std;
#define ll long long
ll res[100010];
int main()
{
	/*在草稿纸上列出算式(竖着)*/
	ios::sync_with_stdio(0);
	int a,n;
	cin>>a>>n;
	if(n==0)
	{
		cout<<0<<endl;//考虑边界情况 
	 } 
	for(int i=n;i>0;i--)
	{
		res[i] += (n-i+1)*a;
	}
	for(int i=1;i<=n;i++)
	{
		res[i+1] +=res[i]/10;
		res[i]%=10;
	}
	int k = n;
	while(res[k+1]!=0)
	{
		k++;
		res[k+1] += res[k]/10;
		res[k]%=10;
	}
	for(int i=k;i>0;i--)
	{
		cout<<res[i];
	}
}