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;
typedef vector<int> vi;
typedef long long ll;
typedef pair<ll,ll> pll;
#define ALL(x) x.begin(),x.end()
#define SIZE(x) (int)x.size()
#define forsn(i,s,n) for(int i=int(s);i<int(n);i++)
#define forn(i,n) for(int i=0;i<int(n);i++)
#define fst first
#define snd second
#define pb push_back
const int MAXN = 2e5;
const int MOD = 1e9+2022;
vi s;
int n, m;
int dp[MAXN][2];
int mul(int x, int y) {
return int(1LL*x*y%MOD);
}
void add(int &x, int y) {
x+=y;
if(x>=MOD) x-=MOD;
}
void compute_dp(int u) {
int l=2*u+1, r=2*u+2;
int aux = mul(dp[l][0], dp[r][1]);
add(aux, mul(dp[l][1], dp[r][0]));
dp[u][1] = dp[u][0] = aux;
aux = mul(mul(2, dp[l][0]), dp[r][0]);
add(dp[u][0], aux);
aux = mul(mul(2, dp[l][1]), dp[r][1]);
add(dp[u][1], aux);
}
void dfs(int u) {
if(u>=n) {
dp[u][s[u-n]] = 1;
dp[u][!s[u-n]] = 0;
return;
}
dfs(2*u+1), dfs(2*u+2);
compute_dp(u);
}
void init(int N, int M, vi /*P*/, vi A) {
s=A, n=N, m=M;
dfs(0);
}
bool lazy[MAXN];
void pass(int u) {
if(!lazy[u]) return;
if(2*u+2<MAXN) {
lazy[2*u+1] ^= 1;
lazy[2*u+2] ^= 1;
}
swap(dp[u][0], dp[u][1]);
lazy[u] = false;
}
void update(int ql, int qr, int nl, int nr, int u) {
pass(u);
if(qr <= nl || nr <= ql) {
return;
}
if(ql <= nl && nr <= qr) {
lazy[u] ^= 1;
pass(u);
return;
}
int nm=(nl+nr)/2;
update(ql,qr,nl,nm,2*u+1);
update(ql,qr,nm,nr,2*u+2);
compute_dp(u);
}
int count_ways(int L, int R) {
update(L-n, R-n+1, 0, m, 0);
return dp[0][1];
}
# | 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... |