#include <bits/stdc++.h>
#define pb push_back
using namespace std;
const int maxN = 2e5 + 5;
const int mod = 1000002022;
void add(int &a, int b){
a += b;
if (a >= mod) a -= mod;
if (a < 0) a += mod;
}
vector <int> adj[maxN], pre[maxN], suf[maxN];
int sz[maxN], val[maxN];
void dfs(int u){
int dmm = adj[u].size();
sz[u] = 1;
for (int i = 0; i < dmm; ++ i){
int v = adj[u][i];
dfs(v);
if (!i) pre[u].pb(sz[v]);
else pre[u].pb(1ll * pre[u].back() * sz[v] % mod);
suf[u].pb(0);
sz[u] += sz[v];
}
for (int i = dmm - 1; i >= 0; -- i){
int v = adj[u][i];
if (i == int(dmm - 1)) suf[u][i] = sz[v];
else suf[u][i] = 1ll * suf[u][i + 1] * sz[v] % mod;
}
}
void get(int u){
int dmm = adj[u].size();
for (int i = 0; i < dmm; ++ i){
int v = adj[u][i];
val[v] = val[u];
if (i > 0) val[v] = 1ll * val[v] * pre[u][i - 1] % mod;
if (i < dmm - 1) val[v] = 1ll * val[v] * suf[u][i + 1] % mod;
}
}
int sum[2][4 * maxN];
bool used[maxN], rev[maxN];
void mer(int s){
sum[0][s] = 0;
sum[1][s] = 0;
for (int i = 0; i <= 1; ++ i){
for (int j = 2 * s; j <= 2 * s + 1; ++ j) add(sum[i][s], sum[i][j]);
}
}
void build(int s, int l, int r, int n){
rev[s] = 0;
if (l == r){
sum[0][s] = used[l] * val[l + n];
sum[1][s] = (1 ^ used[l]) * val[l + n];
return;
}
int mid = (l + r) / 2;
build(2 * s, l, mid, n);
build(2 * s + 1, mid + 1, r, n);
mer(s);
}
int kk, gg;
void init(int N, int M, std::vector<int> P, std::vector<int> A){
kk = N;
gg = M;
for (int i = 1; i < N + M; ++ i) adj[P[i]].pb(i);
dfs(0);
val[0] = 1;
get(0);
for (int i = 0; i < M; ++ i) used[i] = A[i];
build(1, 0, M - 1, N);
}
void down(int s){
if (!rev[s]) return;
for (int i = 2 * s; i <= 2 * s + 1; ++ i) {
rev[i] ^= 1;
swap(sum[0][i], sum[1][i]);
}
rev[s] = 0;
}
void upd(int s, int l, int r, int u, int v){
if (u <= l && r <= v){
rev[s] ^= 1;
swap(sum[0][s], sum[1][s]);
return;
}
int mid = (l + r) / 2;
down(s);
if (mid >= u) upd(2 * s, l, mid, u, v);
if (mid + 1 <= v) upd(2 * s + 1, mid + 1, r, u, v);
mer(s);
}
int count_ways(int L, int R){
upd(1, 0, gg - 1, L - kk, R - kk);
return sum[0][1];
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
20312 KB |
Output is correct |
2 |
Correct |
5 ms |
20312 KB |
Output is correct |
3 |
Correct |
5 ms |
20312 KB |
Output is correct |
4 |
Correct |
6 ms |
20312 KB |
Output is correct |
5 |
Correct |
5 ms |
20312 KB |
Output is correct |
6 |
Correct |
5 ms |
20312 KB |
Output is correct |
7 |
Correct |
5 ms |
20312 KB |
Output is correct |
8 |
Correct |
5 ms |
20312 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
20312 KB |
Output is correct |
2 |
Incorrect |
5 ms |
20312 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '0' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
20312 KB |
Output is correct |
2 |
Correct |
5 ms |
20312 KB |
Output is correct |
3 |
Correct |
5 ms |
20312 KB |
Output is correct |
4 |
Correct |
6 ms |
20312 KB |
Output is correct |
5 |
Correct |
5 ms |
20312 KB |
Output is correct |
6 |
Correct |
5 ms |
20312 KB |
Output is correct |
7 |
Correct |
5 ms |
20312 KB |
Output is correct |
8 |
Correct |
5 ms |
20312 KB |
Output is correct |
9 |
Correct |
5 ms |
20312 KB |
Output is correct |
10 |
Incorrect |
5 ms |
20312 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '0' |
11 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
424 ms |
24408 KB |
1st lines differ - on the 1st token, expected: '431985922', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
424 ms |
24408 KB |
1st lines differ - on the 1st token, expected: '431985922', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
20312 KB |
Output is correct |
2 |
Incorrect |
5 ms |
20312 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '0' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
20312 KB |
Output is correct |
2 |
Correct |
5 ms |
20312 KB |
Output is correct |
3 |
Correct |
5 ms |
20312 KB |
Output is correct |
4 |
Correct |
6 ms |
20312 KB |
Output is correct |
5 |
Correct |
5 ms |
20312 KB |
Output is correct |
6 |
Correct |
5 ms |
20312 KB |
Output is correct |
7 |
Correct |
5 ms |
20312 KB |
Output is correct |
8 |
Correct |
5 ms |
20312 KB |
Output is correct |
9 |
Correct |
5 ms |
20312 KB |
Output is correct |
10 |
Incorrect |
5 ms |
20312 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '0' |
11 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
20312 KB |
Output is correct |
2 |
Correct |
5 ms |
20312 KB |
Output is correct |
3 |
Correct |
5 ms |
20312 KB |
Output is correct |
4 |
Correct |
6 ms |
20312 KB |
Output is correct |
5 |
Correct |
5 ms |
20312 KB |
Output is correct |
6 |
Correct |
5 ms |
20312 KB |
Output is correct |
7 |
Correct |
5 ms |
20312 KB |
Output is correct |
8 |
Correct |
5 ms |
20312 KB |
Output is correct |
9 |
Correct |
5 ms |
20312 KB |
Output is correct |
10 |
Incorrect |
5 ms |
20312 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '0' |
11 |
Halted |
0 ms |
0 KB |
- |