Cod sursa(job #2717240)

Utilizator AlexMariMarinescu Alexandru AlexMari Data 6 martie 2021 19:38:59
Problema Pod Scor 15
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.81 kb
#include<bits/stdc++.h>
using namespace std;
ifstream fin("pod.in");
ofstream fout("pod.out");

const long long MOD=9901;

long long n,m,k,dp[1000005],w[1000005],impossible[1000005];

int main()
{
    fin>>n>>m>>k;
    for(int i=1; i<=m; i++)
    {
       fin>>impossible[i];
        w[impossible[i]]=1;
    }
    dp[0]=1;
    for(int i=1; i<=n; i++)
    {
        if(i<k)
        {
            if(w[i]==0)
                dp[i]=1;
        }
        else
        {
            if(w[i-1]==0)
                {
                    dp[i]+=dp[i-1];
                    dp[i]=dp[i]%MOD;
                }
            if(w[i-k]==0)
                {
                    dp[i]+=dp[i-k];
                    dp[i]=dp[i]%MOD;
                }
        }
    }
    fout<<dp[n];
    return 0;
}