Submission #1009710

#TimeUsernameProblemLanguageResultExecution timeMemory
1009710Double_SlashLinear Garden (IOI08_linear_garden)C++17
95 / 100
42 ms65536 KiB
#include <bits/stdc++.h>

using namespace std;

int n, m, dp[1000001][5][5];  // i, min, max
// L: -1
// P: 1

int main() {
    cin >> n >> m;
    for (int mn = 0; mn <= 2; ++mn) {
        for (int mx = max(mn, 2); mx <= 4; ++mx) {
            dp[n][mn][mx] = 1;
        }
    }
    for (int i = n; i--;) {
        for (int mn = 0; mn <= 2; ++mn) {
            for (int mx = max(mn, 2); mx <= 4; ++mx) {
                // try add L
                if (mn > 0) dp[i][mn][mx] += dp[i + 1][mn - 1][max(2, mx - 1)];
                // try add P
                if (mx < 4) dp[i][mn][mx] += dp[i + 1][min(2, mn + 1)][mx + 1];
                dp[i][mn][mx] %= m;
            }
        }
    }
    string s;
    cin >> s;
    int ans = 1;
    for (int i = 0, mn = 2, mx = 2; i < n; ++i) {
        if (s[i] == 'P') {
            if (mn > 0) {
                ans += dp[i + 1][mn - 1][max(2, mx - 1)];
                ans %= m;
            }
            mn = min(2, mn + 1);
            mx++;
        } else {
            mn--;
            mx = max(2, mx - 1);
        }
    }
    cout << ans << endl;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...