Cod sursa(job #2486212)

Utilizator popashtefan10Popa Stefan popashtefan10 Data 2 noiembrie 2019 14:31:20
Problema Pod Scor 35
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.7 kb
#include <iostream>
#include <cstdio>
#include <algorithm>
#define MOD 9901

using namespace std;

int n, m, k, sc;
int poz[1005], v[50];

struct matrice {
  int a[25][25];

  matrice() {
    for(int i = 0; i <= k + 2; i++)
      for(int j = 0; j <= k + 2; j++)
        a[i][j] = 0;
  }

  void afisare() {
    for(int i = 1; i <= k; i++) {
      for(int j = 1; j <= k; j++)
        printf("%d ", a[i][j]);
      printf("\n");
    }
  }

  matrice operator* (const matrice x) {
    matrice rez;

    for(int i = 1; i <= k; i++)
      for(int j = 1; j <= k; j++)
        for(int p = 1; p <= k; p++)
          rez.a[i][j] =  (rez.a[i][j] + (a[i][p] * x.a[p][j]) % MOD) % MOD;
    return rez;
  }

};

matrice getM() {
  matrice M = matrice();

  for(int i = 1; i < k; i++)
    M.a[i + 1][i] = 1;
  M.a[1][k] = M.a[k][k] = 1;

  return M;
}

matrice ide() {
  matrice x;

  for(int i = 1; i <= k; i++)
    x.a[i][i] = 1;
  return x;
}

matrice fast_pow(matrice M, int pow) {
  matrice rez = ide();

  if(pow == 0)
    return rez;
  if(pow == 1)
    return M;
  while(pow > 1) {
    if(pow % 2 == 1)
      rez = rez * M;
    M = M * M;
    pow /= 2;
  }
  return M * rez;
}

int main() {
  freopen("pod.in", "r", stdin);
  freopen("pod.out", "w", stdout);

  scanf("%d %d %d", &n, &m, &k);
  for(int i = 1; i <= m; i++)
    scanf("%d", &poz[i]);
  sort(poz + 1, poz + m + 1);
  poz[m + 1] = n + 1;

  matrice sol = matrice();
  sol.a[1][k] = 1;
  for(int sc = 1; sc <= m + 1; sc++) {
    matrice M = getM();
    M = fast_pow(M, poz[sc] - poz[sc - 1]);
    sol = sol * M;
    sol.a[1][k] = 0;
  }

  printf("%d", sol.a[1][k - 1]);

  return 0;
}