Submission #1251121

#TimeUsernameProblemLanguageResultExecution timeMemory
1251121rhm_ganTrener (COCI20_trener)C++20
110 / 110
557 ms5816 KiB
#include <bits/stdc++.h>
using namespace std;

#ifdef LOCAL
#include "debug.h"
#else
#define dbg(...) 42
#endif

const int mod = 1e9 + 7;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, k;
    cin >> n >> k;

    vector<vector<string>> s(n, vector<string>(k));
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < k; j++) {
            cin >> s[i][j];
        }
    }

    vector<vector<int>> dp(n, vector<int>(k));
    fill(dp[0].begin(), dp[0].end(), 1);

    for (int i = 1; i < n; i++) {
        for (int j = 0; j < k; j++) {
            string s1 = s[i][j].substr(0, i);
            string s2 = s[i][j].substr(1, i);
            for (int r = 0; r < k; r++) {
                if (s[i - 1][r] == s1 || s[i - 1][r] == s2) {
                    dp[i][j] = (dp[i][j] + dp[i - 1][r]) % mod;
                }
            }
        }
    }

    int res = 0;
    for (int i = 0; i < k; i++) {
        res = (res + dp[n - 1][i]) % mod;
    }

    cout << res << '\n';

    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...