PTA 数列求和加强版代码

PTA 数列求和加强版代码

使用string组合字符串输出再逆序输出会超时，直接用char数组保存结果反向输出，char数组申请100001个空间即可

#include<iostream>
#include<string>
#include<algorithm>
using namespace std;

int main(void){
    int A,N;
    cin >> A>> N;
    int sum = 0;
    int flag = 0;
    char result[200000];
    
    char add1;
    if(N==0) {cout<<0; return 0;}
    int epoch = 1;
    for(int i=0;i<N;i++){
        int temp=0;
        temp = A*(N-i)+flag;
        flag = temp/10;
        add1 = temp%10 + '0';
        result[i]=add1;
    }
    char b;
    if(flag != 0) { b= flag + '0';
     result[N]=b;N++; }
    for(int i=N-1;i>=0;i--){
        cout<<result[i];
    }
    return 0;
}