Cod sursa(job #1828753)

Utilizator minut1Baies Cosmin minut1 Data 13 decembrie 2016 20:48:32
Problema Lista lui Andrei Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.43 kb
#include <fstream>

using namespace std;

ifstream cin ("div3.in");
ofstream cout ("div3.out");

int n,d[1001][3];

int main()
{
    int k,p,m,c,s=0;
    cin>>n>>p;
    m=4001;
    for(int i=1; i<=n; i++)
    {
        cin>>c;
        d[1][c%3]++;
    }
    for(int i=2; i<=p; i++)
        for(int j=0; j<3; j++)
            for(k=0; k<3; k++)
                d[i][j]=(d[i][j]+d[i-1][k] * d[1][(j-k+3)%3])%m;
    cout<<d[p][0];
    return 0;
}