oj.uz

답안 #580288

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
580288 2022-06-21T03:45:39 Z tranxuanbach Linear Garden (IOI08_linear_garden) C++17
100 / 100
 21 ms 14080 KB 
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

#define endl '\n'
#define fi first
#define se second
#define For(i, l, r) for (auto i = (l); i < (r); i++)
#define ForE(i, l, r) for (auto i = (l); i <= (r); i++)
#define FordE(i, l, r) for (auto i = (l); i >= (r); i--)
#define Fora(v, a) for (auto v: (a))
#define bend(a) (a).begin(), (a).end()
#define isz(a) ((signed)(a).size())

using ll = long long;
using ld = long double;
using pii = pair <int, int>;
using vi = vector <int>;
using vpii = vector <pii>;
using vvi = vector <vi>;

const int N = 1e6 + 5;
int mod;

inline void sadd(int &x, int y){
    if ((x += y) >= mod){
        x -= mod;
    }
}

inline int add(int x, int y){
    if ((x += y) >= mod){
        x -= mod;
    }
    return x;
}

inline void ssub(int &x, int y){
    if ((x -= y) < 0){
        x += mod;
    }
}

inline int sub(int x, int y){
    if ((x -= y) < 0){
        x += mod;
    }
    return x;
}

int n;
string s;

int dp[3][N];

signed main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    // freopen("KEK.inp", "r", stdin);
    // freopen("KEK.out", "w", stdout);
    cin >> n >> mod;

    dp[-1 + 1][0] = 1;
    dp[0 + 1][0] = 1;
    dp[1 + 1][0] = 1;
    ForE(i, 1, n){
        ForE(d, -1, 1){
            if (d >= 0){
                sadd(dp[d + 1][i], dp[d - 1 + 1][i - 1]);
            }
            if (d <= 0){
                sadd(dp[d + 1][i], dp[d + 1 + 1][i - 1]);
            }
        }
    }

    cin >> s;
    int d = 0, mxd = 0, mnd = 0, ans = 1;
    For(i, 0, n){
        if (s[i] == 'P'){
            d--;
            int tmxd = max(mxd, d), tmnd = min(mnd, d);
            if (-2 <= tmnd and tmxd <= 0){
                sadd(ans, dp[(d + 1) + 1][n - i - 1]);
            }
            if (-1 <= tmnd and tmxd <= 1){
                sadd(ans, dp[d + 1][n - i - 1]);
            }
            if (0 <= tmnd and tmxd <= 2){
                sadd(ans, dp[(d - 1) + 1][n - i - 1]);
            }
            if (-1 <= tmnd and tmxd <= 0){
                ssub(ans, 1);
            }
            if (0 <= tmnd and tmxd <= 1){
                ssub(ans, 1);
            }
            d++;
        }
        if (s[i] == 'L'){
            d--;
        }
        else{
            d++;
        }
        mxd = max(mxd, d);
        mnd = min(mnd, d);
    }
    cout << ans << endl;
}

/*
==================================================+
INPUT:                                            |
--------------------------------------------------|

--------------------------------------------------|
==================================================+
OUTPUT:                                           |
--------------------------------------------------|

--------------------------------------------------|
==================================================+
*/

Subtask #1
2.0 / 2.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #2
2.0 / 2.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #3
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 340 KB Output is correct

Subtask #4
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #5
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #6
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 340 KB Output is correct

Subtask #7
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 340 KB Output is correct

Subtask #8
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 340 KB Output is correct

Subtask #9
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #10
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #11
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct

Subtask #12
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 340 KB Output is correct
2 Correct 0 ms 340 KB Output is correct

Subtask #13
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 340 KB Output is correct
2 Correct 1 ms 340 KB Output is correct

Subtask #14
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 340 KB Output is correct
2 Correct 1 ms 328 KB Output is correct
3 Correct 1 ms 340 KB Output is correct

Subtask #15
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 596 KB Output is correct

Subtask #16
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 596 KB Output is correct

Subtask #17
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 1108 KB Output is correct

Subtask #18
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 1364 KB Output is correct

Subtask #19
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 6 ms 4576 KB Output is correct

Subtask #20
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 5644 KB Output is correct

Subtask #21
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 11 ms 6900 KB Output is correct

Subtask #22
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 13 ms 8680 KB Output is correct

Subtask #23
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 21 ms 13084 KB Output is correct
2 Correct 21 ms 14080 KB Output is correct
3 Correct 19 ms 14016 KB Output is correct