Cod sursa(job #2432561)

Utilizator AlexPop28Pop Alex-Nicolae AlexPop28 Data 24 iunie 2019 12:49:48
Problema Pod Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.54 kb
#include <bits/stdc++.h>
#define DEBUG(x) cerr << (#x) << ": " << (x) << '\n'

using namespace std;

const int MOD = 9901;

struct Matrix {
  int n;
  vector<vector<int>> a;
  Matrix(int n) : n(n), a(n, vector<int>(n)) {}
  vector<int>& operator[](const int pos) {
    return a[pos];
  }
  Matrix operator*(Matrix &b) const {
    Matrix res(n);
    for (int i = 0; i < n; ++i) {
      for (int k = 0; k < n; ++k) {
        for (int j = 0; j < n; ++j) {
          res[i][j] += a[i][k] * b[k][j] % MOD;
          if (res[i][j] >= MOD) res[i][j] -= MOD;
        }
      }
    }
    return res;
  }
  Matrix toPow(int e) {
    Matrix res = *this, b = *this;
    --e;
    while (e) {
      if (e & 1)
        res = res * b;
      b = b * b;
      e >>= 1;
    }
    return res;
  }
};

int main() {
  freopen("gedit.in", "r", stdin);
  freopen("gedit.out", "w", stdout);
  ios::sync_with_stdio(0);
  cin.tie(0);

  int n, m, k; cin >> n >> m >> k;
  vector<int> v(m);
  for (int i = 0; i < m; ++i) {
    cin >> v[i];
  }
  Matrix tr(k);
  for (int i = 0; i < k - 1; ++i) {
    tr[i][i + 1] = 1;
  }
  tr[k - 1][0] = tr[k - 1][k - 1] = 1;
  
  Matrix dp(k);
  dp[k - 1][0] = 1; // dp = {{dp[-i],.., 0},..., {dp[0], 0}}
  
  int last = 0;
  for (int i = 0; i < m; ++i) {
    int e = v[i] - last;
    dp = tr.toPow(e) * dp;
    dp[k - 1][0] = 0; // dp[v[i]] = 0
    last = v[i];
  }
  dp = tr.toPow(n - last) * dp;
  cout << dp[k - 1][0] << endl;

#ifdef LOCAL_DEFINE
  cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
  return 0;
}