Cod sursa(job #1886796)

Utilizator BlackLordFMI Alex Oprea BlackLord Data 21 februarie 2017 10:01:35
Problema Pod Scor 15
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.17 kb
#include <fstream>
#include <algorithm>

using namespace std;

ifstream fin ("pod.in");
ofstream fout ("pod.out");

const int N = 50, mod = 9901, M = 1010;

int n, m, k, v[M], d[N], i, j, c, l;

int main() {
    fin >> n >> m >> k;
    for (i = 1; i <= m; ++i) {
        fin >> v[i];
    }
    sort (v + 1, v + m + 1);
    j = 1;
    d[0] = 1;
    for (i = 1; i < k; ++i) {
        if (i == v[j]) {
            d[i] = 0;
            ++j;
        } else {
            d[i] = d[i - 1];
        }
    }
    if (k == d[j]) {
        d[k] = 0;
        j++;
    } else {
        d[k] = d[k - 1] + 1;
    }
    for (i = k + 1, l = k + 1; i <= n; ++i, ++l) {
        if (i % k == 1) {
            if ((i / k) % 2 == 0) {
                c = -1;
                l = 1;
            } else {
                c = 1;
            }
        }
        if (i == v[j]) {
            d[l] = 0;
            ++j;
        } else {
            if (l == 1) {
                d[l] = (d[2 * k] + d[l - k * c]) % mod;
            } else {
                d[l] = (d[l - 1] + d[l - k * c]) % mod;
            }
        }
    }
    fout << d[l - 1] << "\n";
    return 0;
}