Submission #1036584

#TimeUsernameProblemLanguageResultExecution timeMemory
1036584VMaksimoski008Linear Garden (IOI08_linear_garden)C++17
65 / 100
52 ms65536 KiB
#include <bits/stdc++.h>

#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
//#define int long long

using namespace std;

using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;

const int LOG = 20;
const int maxn = 1e5 + 5;

ll n, mod, ans = 0;
vector<ll> vec;
string s;

int dp[60005][5][5][5][2];

int f(int pos, int mn, int mx, int curr, int under) {
    if(mx - mn > 2) return 0;
    if(mx > 4 || mn < 0) return 0;
    if(pos == n + 1) return under;
    if(dp[pos][mn][mx][curr][under] != -1) return dp[pos][mn][mx][curr][under];

    ll ans = 0;

    for(int i=-1; i<=1; i++) {
        if(i == 0) continue;
        if(!under && i > vec[pos]) continue;
        int nmn = min(mn, curr + i);
        int nmx = max(mx, curr + i);
        int nu = under | (i < vec[pos]);
        ans = (ans + f(pos+1, nmn, nmx, curr + i, nu)) % mod;
    }

    return dp[pos][mn][mx][curr][under] = ans;
}

signed main() {
    memset(dp, -1, sizeof(dp));
    cin >> n >> mod >> s;
    vec.push_back(0);

    for(int i=0; i<n; i++) {
        if(s[i] == 'L') vec.push_back(-1);
        else vec.push_back(1);
    }

    ll ans = f(1, 2, 2, 2, 0);
    cout << (ans + 1) % mod << '\n';
    return 0;
}
#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...