diff options
-rw-r--r-- | 20/dec/plat/spaceship.cpp | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/20/dec/plat/spaceship.cpp b/20/dec/plat/spaceship.cpp new file mode 100644 index 0000000..d8d7849 --- /dev/null +++ b/20/dec/plat/spaceship.cpp @@ -0,0 +1,49 @@ +#include <bits/stdc++.h> +#define f first +#define s second +using namespace std; +using ll = long long; +using ii = pair<int, int>; +constexpr int MX = 65, MOD = 1e9+7; + +ll bs[MX], s[MX], bt[MX], t[MX], L[2*MX], R[2*MX], G[MX][MX], DP[MX][2*MX][2*MX]; + +int main() { + if (fopen("in", "r")) freopen("in", "r", stdin), freopen("out", "w", stdout); + ios_base::sync_with_stdio(0), cin.tie(0); + + int N, K, Q; cin >> N >> K >> Q; + for (int i = 0; i < N; ++i) { + string s; cin >> s; + for (int j = 0; j < N; ++j) G[i][j] = s[j] == '1'; + } + for (int i = 0; i < Q; ++i) + cin >> bs[i] >> s[i] >> bt[i] >> t[i], --bs[i], --s[i], --bt[i], --t[i]; + + for (int i = 0; i < K; ++i) { + for (int j = 0; j < N; ++j) { + for (int k = 0; k < N+Q; ++k) L[k] = k == j; + for (int k = 0; k < Q; ++k) + if (i == bs[k] && j == s[k]) L[N+k] = 1; + + for (int k = 0; k < N+Q; ++k) + for (int l = 0; l < N; ++l) + if (i && G[l][j]) (L[k] += DP[i-1][k][l]) %= MOD; + + for (int k = 0; k < N+Q; ++k) R[k] = k == j; + for (int k = 0; k < Q; ++k) + if (i == bt[k] && j == t[k]) R[N+k] = 1; + + for (int k = 0; k < N+Q; ++k) + for (int l = 0; l < N; ++l) + if (i && G[j][l]) (R[k] += DP[i-1][l][k]) %= MOD; + + for (int k = 0; k < N+Q; ++k) + for (int l = 0; l < N+Q; ++l) + (DP[i][k][l] += L[k]*R[l]) %= MOD; + } + memcpy(DP[i+1], DP[i], sizeof DP[i]); + } + + for (int i = 0; i < Q; ++i) cout << DP[K][N+i][N+i] << '\n'; +} |