Cod sursa(job #1551457)

Utilizator bciobanuBogdan Ciobanu bciobanu Data 15 decembrie 2015 21:51:45
Problema Cifre Scor 100
Compilator c Status done
Runda Arhiva de probleme Marime 1.98 kb
#include <stdio.h>
#include <string.h>
#include <assert.h>

#define MAX_N 12
#define SMECHERIE 5000000
#define LIMIT 10000

int dig, freq;

int D[MAX_N][MAX_N];

int cntDigs( int X ) {
  int tmp, fr = 0;
  do {
    tmp = X / 10;
    fr += ( X == dig + (tmp<<1) + (tmp<<3) );
    X = tmp;
  } while ( X && (fr != freq) );
  return fr;
}

int memo( int N, int f ) {
  if ( N == 0 ) {
    return ( f >= freq );
  }

  if ( D[N][f] != -1 ) {
    return D[N][f];
  }

  int i, q;

  q = 0;
  for ( i = 0; i < 10; i++ ) {
    q += memo( N - 1, f + ( dig == i ) );
  }
  D[N][f] = q;

  return q;
}

int DP( int X ) {
  char buf[MAX_N];
  int N, i, j, currDig, q, fr;

  for ( i = 0; i < MAX_N; i++ )
    for ( j = 0; j < MAX_N; j++ )
      D[i][j] = -1;

  sprintf( buf, "%d", X );
  N = strlen( buf );

  q = fr = 0;

  for ( i = 0; i < N; i++ ) {
    currDig = buf[i] - '0';
    for ( j = 0; j < currDig; j++ )
      q += memo( N - i - 1, fr + ( dig == j ) );
    fr += ( currDig == dig );
  }
  return q;
}

int main(void) {
  FILE *fin, *fout;
  int A, B, i, tmp;
  int numarator;

  fin = fopen( "cifre.in", "r" );
  fscanf( fin, "%d%d%d%d", &A, &B, &dig, &freq );
  fclose( fin );

  fout = fopen( "cifre.out", "w" );

  if ( freq == 0 ) {
    fputs( "1\n", fout );
  } else {
    if ( dig == 0 ) {
      numarator = 0;
      i = A;
      while ( A - A / LIMIT * LIMIT ) {
        numarator += ( cntDigs( A++ ) == freq );
      }
      while ( B - A >= LIMIT ) {
        tmp = cntDigs( A / LIMIT );
        numarator += 6561 & -( tmp == freq );
        numarator += 2916 & -( tmp + 1 >= freq );
        numarator += 487 & -( tmp + 2 >= freq );
        numarator += 35 & -( tmp + 3 >= freq );
        numarator += 1 & -( tmp + 4 >= freq );
        A += LIMIT;
      }
      while ( A <= B ) {
        numarator += ( cntDigs( A++ ) == freq );
      }
      A = i;
    } else if ( dig != 0 ) {
      numarator = DP(B+1) - DP(A);
    }
    fprintf( fout, "%lf\n", 1.00 * numarator / (B-A+1) );
  }
  fclose( fout );
  return 0;
}