Cod sursa(job #350636)

Utilizator prisonbreakMichael Scofield prisonbreak Data 25 septembrie 2009 09:18:41
Problema Sarpe Scor 100
Compilator cpp Status done
Runda info.conc.sept.2 Marime 1.77 kb
#include <cstdio>
#include <string>
#define DIM 10000000
using namespace std;
int i,a[DIM],n,b[DIM],d[DIM],c[DIM];
char t[DIM];
void mul(int A[], int B)
{
      int i, t = 0;
      for (i = 1; i <= A[0] || t; i++, t /= 10)
              A[i] = (t += A[i] * B) % 10;
      A[0] = i - 1;
}
void mulm(int A[], int B[])
{
      int i, j, t, C[100000];
      memset(C, 0, sizeof(C));
      for (i = 1; i <= A[0]; i++)
      {
              for (t=0, j=1; j <= B[0] || t; j++, t/=10)
                      C[i+j-1]=(t+=C[i+j-1]+A[i]*B[j])%10;
              if (i + j - 2 > C[0]) C[0] = i + j - 2;
      }
      memcpy(A, C, sizeof(C));
}
void sub(int A[], int B[])
{
      int i, t = 0;
      for (i = 1; i <= A[0]; i++)
              A[i] += (t = (A[i] -= B[i] + t) < 0) * 10;
      for (; A[0] > 1 && !A[A[0]]; A[0]--);
}
void add(int A[], int B[])
{
      int i, t = 0;
      for (i=1; i<=A[0] || i<=B[0] || t; i++, t/=10)
              A[i] = (t += A[i] + B[i]) % 10;
      A[0] = i - 1;
}
void solve()
{   
    int j=0;
    for (i=n-1; i>=0; --i) b[++j]=d[j]=a[j]=(t[i]-'0');
    a[0]=d[0]=b[0]=n;
    c[0]=1;c[1]=1;
    //for (i=b[0]; i>=1; --i) printf("%d",b[i]);printf("\n");
    sub (b,c); // N-1 
    //for (i=b[0]; i>=1; --i) printf("%d",b[i]);
    //printf("\n");
    c[0]=1;c[1]=2;
    sub (d,c); // N-2
   mul(a,4); // 4*N
    
    mulm (b,d);// (N-1) (N-2) 
    
    mul (b,2); // 2* (N-1) (N-2)
    
    add (a,b);// 4*N + 2* (N-1) (N-2)
    for (i=a[0]; i>=1; i--) printf("%d",a[i]);printf("\n");
}   
int main()
{
    freopen ("sarpe.in","r",stdin);
    freopen ("sarpe.out","w",stdout);
    scanf ("%s\n",&t);
    n = strlen (t);
    if (n==1 && t[0]=='1') printf("2"); 
    else solve();
return 0;
}