Cod sursa(job #1447127)

Utilizator akaprosAna Kapros akapros Data 3 iunie 2015 18:36:20
Problema Diviz Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.6 kb
#include <cstdio>
#include <algorithm>
#include <cstring>
#define Nmax 205
#define Kmax 105
#define mod 30103
using namespace std;
int n, i, j, k, a, b, r, cif;
int d[2][Nmax][Kmax];
int sol, v[Nmax];
int first[12][Nmax];
char s[Nmax];
int main()
{
    freopen("diviz.in", "r", stdin);
    freopen("diviz.out", "w", stdout);
    scanf("%d %d %d\n", &k, &a, &b);
    gets(s + 1);
    n = strlen(s + 1);

    for (i = 1; i <= n; ++ i)
      v[i] = s[i] - '0';

    for (i = n; i >= 1; -- i)
    for (j = 0; j <= 9; ++ j)
    if (v[i] != j)
    first[j][i] = first[j][i + 1];
    else first[j][i] = i;

    for (j = 0; j <= 9; ++ j)
    if (first[j][1])
    {
        if (j == 0 ) {if (a == 1) ++ sol; continue;}
          d[1][first[j][1]][j % k] = 1;
        if (a == 1 && j % k == 0)
          sol = (sol + d[1][first[j][1]][j % k]) % mod;
    }
    memset(d, 0, sizeof(d));

    for (i = 1; i <= n; ++ i)
    if (v[i] != 0)
    {
        ++ d[1][i][v[i] % k];
    }

    for (i = 1; i < b; ++ i)
    {
        for (j = 1; j <= n; ++ j)
         for (cif = 0; cif <= 9; ++ cif)
           if (first[cif][j + 1])
             for (r = 0; r < k; ++ r)
           {
               d[(i + 1) % 2][first[cif][j + 1]][(r * 10 + cif) % k] =
               ( d[i % 2][j][r] + d[(i + 1) % 2][first[cif][j + 1]][(r * 10 + cif) % k]) % mod;

           }
        if (i + 1>= a)
        {
            for (j = 1; j <= n; ++ j)
            sol = (sol + d[(i + 1) % 2][j][0]) % mod;
        }

        memset(d[i % 2], 0, sizeof(d[i % 2]));

    }
    printf("%d", sol);

    return 0;
}