Cod sursa(job #3219870)

Utilizator Dia3141Costea Diana Stefania Dia3141 Data 1 aprilie 2024 17:42:25
Problema Sarpe Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.41 kb
#include <fstream>
#include <cstring>
using namespace std;
ifstream cin("sarpe.in");
ofstream cout("sarpe.out");
char s[2005];
int nr[2005],v[2005],r[2005],r2[2005];
void produs(int a[],int b,int c[]){
    int r=0;
    for(int i=1;i<=a[0];i++){
        c[i]=r+a[i]*b;
        r=c[i]/10;
        c[i]%=10;
    }
    c[0]=a[0];
    while(r!=0){
        c[++c[0]]=r%10;
        r/=10;
    }
}
void produs2(int a[],int b[],int c[]){
    c[0]=a[0]+b[0]-1;
    for(int i=1;i<=b[0];i++)
        for(int j=1;j<=a[0];j++)
            c[j+i-1]+=a[j]*b[i];
    int r=0;
    for(int i=1;i<=c[0];i++){
        c[i]+=r;
        r=c[i]/10;
        c[i]%=10;
    }
    if(r!=0)
        c[++c[0]]=r;
}
void scadere(int a[]){
    int poz=1;
    while(a[poz]==0)
        a[poz++]=9;
    a[poz]--;

}
void suma(int a[],int b[]){
    int n=max(a[0],b[0]),r=0;
    a[0]=n;
    for(int i=1;i<=n;i++){
        a[i]+=b[i]+r;
        r=a[i]/10;
        a[i]%=10;
    }
    if(r!=0)
        a[++a[0]]=r;
}
void afisare(int a[]){
    for(int i=a[0];i>0;i--)
        cout<<a[i];
}
int main()
{
    cin>>s;
    int n=strlen(s);
    for(int i=n-1;i>=0;i--)
        nr[++nr[0]]=s[i]-'0';
    if(nr[0]==1&&nr[1]==1){
        cout<<2;
        return 0;
    }
    produs(nr,4,r);
    scadere(nr);
    produs(nr,2,v);
    scadere(nr);
    produs2(v,nr,r2);
    suma(r,r2);
    afisare(r);
    return 0;
}