This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
int dp[500001][5][3][3];
int n,m;
int solve(int i,int cnt,int ma,int mi){
if(i==n){
return 1;
}
if(dp[i][cnt][ma][mi]!=-1)return dp[i][cnt][ma][mi];
int c1 = 0;
{
int ne = cnt+1;
int r = 2+ma;
int l = 2-mi;
if(abs(ne-r)<=2&&abs(ne-l)<=2){
int nma = max(ma,ne-2);
int nmi = max(mi,2-ne);
c1+=solve(i+1,ne,nma,nmi);
c1%=m;
}
}
{
int ne = cnt-1;
int r = 2+ma;
int l = 2-mi;
if(abs(ne-r)<=2&&abs(ne-l)<=2){
int nma = max(ma,ne-2);
int nmi = max(mi,2-ne);
c1+=solve(i+1,ne,nma,nmi);
c1%=m;
}
}
return dp[i][cnt][ma][mi] = c1;
}
int main(){
cin>>n>>m;
string s;cin>>s;
memset(dp,-1,sizeof dp);
int cnt = 2;
int num = 1 ;
int ma =0 , mi = 0;
for(int i = 0;i<s.size();i++){
if(s[i]=='P'){
int ne = cnt-1;
int r = 2+ma;
int l = 2-mi;
if(abs(ne-r)<=2&&abs(ne-l)<=2){
int nma = max(ma,ne-2);
int nmi = max(mi,2-ne);
num+=solve(i+1,ne,nma,nmi);
num%=m;
}
}
if(s[i]=='P')cnt++;
else cnt--;
ma = max(ma,cnt-2);
mi = max(mi,2-cnt);
}
cout<<num<<"\n";
}
Compilation message (stderr)
linear_garden.cpp: In function 'int main()':
linear_garden.cpp:43:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
43 | for(int i = 0;i<s.size();i++){
| ~^~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |