Cod sursa(job #2486540)

Utilizator popashtefan10Popa Stefan popashtefan10 Data 3 noiembrie 2019 08:29:56
Problema Pod Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.83 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;
  }

} puteri[55], M, I;

matrice getM() {
  matrice mat = matrice();

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

  return mat;
}

matrice ide() {
  matrice x;

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

matrice fast_pow(int pow) {
  matrice rez = I;

  if(pow == 0)
    return rez;
  if(pow == 1)
    return M;
  for(int i = 0; (1 << i) <= pow; i++)
    if(pow & (1 << i))
      rez = rez * puteri[i];
  return 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;

  M = getM();
  I = ide();
  puteri[0] = M;
  for(int i = 1; i <= 50; i++)
    puteri[i] = puteri[i - 1] * puteri[i - 1];

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

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

  return 0;
}