a/IOI
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add garden.cpp

Browse source
This commit is contained in:
Anthony Wang 2020-10-06 17:29:00 -05:00
parent c4e19326d1
commit ac539fe4aa
Signed by: a
GPG key ID: 6FD3502572299774
1 changed files with 31 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
08/garden.cpp Normal file
View file

@ -0,0 +1,31 @@
#include <bits/stdc++.h>
#define f first
#define s second
using namespace std;
using ll = long long;
using ii = pair<int, int>;
constexpr int MX = 1e6+5;
int p2[MX] = { 1 };
int main() {
if (fopen("in", "r")) freopen("in", "r", stdin), freopen("out", "w", stdout);
ios_base::sync_with_stdio(0), cin.tie(0);
int N, M; cin >> N >> M;
string S; cin >> S;
for (int i = 0; i < N; ++i) p2[i+1] = 2*p2[i]%M;
int ans = 1, cnt = 0, l = 0, h = 0;
for (int i = 0; i < N; ++i) {
if (S[i] == 'P') {
int a = min(cnt-1, l), b = max(cnt-1, h);
if (abs(b-a) == 1) (ans += p2[(N-i-1)/2]+p2[(N-i)/2]-1) %= M;
else if (abs(b-a) == 2) (ans += p2[(N-i-(cnt != a))/2]) %= M;
}
cnt += (S[i] == 'P' ? 1 : -1);
l = min(cnt, l), h = max(cnt, h);
}
cout << ans;
}