Submission #503833

#	Time	Username	Problem	Language	Result	Execution time	Memory
503833		tabr	Linear Garden (IOI08_linear_garden)	C++17	100 / 100	135 ms	2400 KiB

linear_garden

#include <bits/stdc++.h>
using namespace std;
#ifdef tabr
#include "library/debug.cpp"
#else
#define debug(...)
#endif

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n, mod;
    cin >> n >> mod;
    string s;
    cin >> s;
    int dp[2][7][7][7] = {};
    int mn = 3;
    int mx = 3;
    int now = 3;
    for (int i = 0; i < n; i++) {
        int p = i & 1;
        for (int x = 1; x < 7; x++) {
            for (int y = x + 1; y < min(6, x + 3); y++) {
                for (int z = x; z <= y; z++) {
                    if (y - min(z - 1, x) < 3) {
                        dp[p ^ 1][z - 1][min(z - 1, x)][y] += dp[p][z][x][y];
                    }
                    if (min(z + 1, y) - x < 3) {
                        dp[p ^ 1][z + 1][x][max(z + 1, y)] += dp[p][z][x][y];
                    }
                }
            }
        }
        if (s[i] == 'P') {
            if (mx - min(mn, now - 1) < 3) {
                dp[p ^ 1][now - 1][min(mn, now - 1)][mx]++;
            }
            now++;
            mx = max(mx, now);
        } else {
            now--;
            mn = min(mn, now);
        }
        for (int x = 1; x < 7; x++) {
            for (int y = x + 1; y < min(6, x + 3); y++) {
                for (int z = x; z <= y; z++) {
                    while (dp[p ^ 1][z][x][y] >= mod) {
                        dp[p ^ 1][z][x][y] -= mod;
                    }
                    dp[p][z][x][y] = 0;
                }
            }
        }
    }
    int ans = 1;
    for (int i = 0; i < 7; i++) {
        for (int j = i + 1; j < min(7, i + 3); j++) {
            for (int k = i; k <= j; k++) {
                ans += dp[n & 1][k][i][j];
                if (ans >= mod) {
                    ans -= mod;
                }
            }
        }
    }
    cout << ans << '\n';
    return 0;
}

• Subtask 1: 2 / 2
• Subtask 2: 2 / 2
• Subtask 3: 4 / 4
• Subtask 4: 4 / 4
• Subtask 5: 4 / 4
• Subtask 6: 4 / 4
• Subtask 7: 4 / 4
• Subtask 8: 4 / 4
• Subtask 9: 4 / 4
• Subtask 10: 4 / 4
• Subtask 11: 4 / 4
• Subtask 12: 5 / 5
• Subtask 13: 5 / 5
• Subtask 14: 5 / 5
• Subtask 15: 5 / 5
• Subtask 16: 5 / 5
• Subtask 17: 5 / 5
• Subtask 18: 5 / 5
• Subtask 19: 5 / 5
• Subtask 20: 5 / 5
• Subtask 21: 5 / 5
• Subtask 22: 5 / 5
• Subtask 23: 5 / 5