#include "circuit.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
using ll = long long;
const int N = 3e5 + 10;
const int MOD = 1000002022;
int n, m;
int A[N];
vector<int> g[N];
int sz[N];
void dfs1(int x){
sz[x] = g[x].size();
for(auto u: g[x]){
dfs1(u);
sz[x] *= sz[u];
sz[x] %= MOD;
}
if(!sz[x])sz[x] = 1;
}
int X[N];
void dfs2(int x,int w){
X[x] = w;
int k = g[x].size();
int pref[k], suff[k];
for(int i = 0;i<k;i++){
pref[i] = sz[g[x][i]];
if(i) pref[i] = (pref[i-1] * pref[i]) % MOD;
}
for(int i = k - 1;i>=0;i--){
suff[i] = sz[g[x][i]];
if(i != k - 1)suff[i] = (suff[i] * suff[i+1]) % MOD;
}
for(int i = 0;i<k;i++){
int prod = w;
if(i)prod = (prod * pref[i-1]) % MOD;
if(i + 1 < k)prod = (prod * suff[i+1]) % MOD;
dfs2(g[x][i], prod);
}
}
array<ll, 2> t[4 * N];
bool lazy[4 * N];
void build(int k,int tl,int tr){
if(tl == tr){
t[k] = {X[tl], 0};
if(A[tl])swap(t[k][0], t[k][1]);
return;
}
int tm = (tl + tr) / 2;
build(k *2, tl, tm);
build(k*2+1, tm + 1, tr);
t[k][0] = (t[k*2][0] + t[k*2+1][0]) % MOD;
t[k][1] = (t[k*2][1] + t[k*2+1][1]) % MOD;
}
void push(int k, int tl,int tr){
if(!lazy[k])return;
swap(t[k][0], t[k][1]);
if(tl != tr){
lazy[k*2] ^= 1;
lazy[k*2+1] ^= 1;
}
lazy[k] = 0;
}
void update(int k,int tl,int tr,int l,int r){
push(k,tl, tr);
if(r < tl || tr < l)return;
if(l <= tl && tr <= r){
lazy[k] ^= 1;
push(k,tl, tr);
return;
}
int tm = (tl + tr) / 2;
update(k * 2, tl, tm, l, r);
update(k*2+1, tm+1,tr, l, r);
t[k][0] = (t[k*2][0] + t[k*2+1][0]) % MOD;
t[k][1] = (t[k*2][1] + t[k*2+1][1]) % MOD;
}
void init(int nn, int mm, std::vector<int> P, std::vector<int> a) {
n = nn;
m = mm;
for(int i = 1;i < n + m;i++){
g[P[i]].pb(i);
}
for(int i = n;i<n + m;i++)A[i] = a[i - n];
dfs1(0);
dfs2(0, 1);
build(1, n, n + m - 1);
}
int count_ways(int L, int R) {
update(1, n, n + m - 1, L, R);
push(1, n, n + m - 1);
return t[1][1];
}
//Te molam da raboti !!
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
13144 KB |
Output is correct |
2 |
Correct |
2 ms |
13144 KB |
Output is correct |
3 |
Correct |
2 ms |
13144 KB |
Output is correct |
4 |
Correct |
3 ms |
13144 KB |
Output is correct |
5 |
Correct |
3 ms |
13144 KB |
Output is correct |
6 |
Correct |
3 ms |
13140 KB |
Output is correct |
7 |
Correct |
2 ms |
13144 KB |
Output is correct |
8 |
Correct |
2 ms |
13144 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
13144 KB |
Output is correct |
2 |
Incorrect |
3 ms |
13144 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '83005428' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
13144 KB |
Output is correct |
2 |
Correct |
2 ms |
13144 KB |
Output is correct |
3 |
Correct |
2 ms |
13144 KB |
Output is correct |
4 |
Correct |
3 ms |
13144 KB |
Output is correct |
5 |
Correct |
3 ms |
13144 KB |
Output is correct |
6 |
Correct |
3 ms |
13140 KB |
Output is correct |
7 |
Correct |
2 ms |
13144 KB |
Output is correct |
8 |
Correct |
2 ms |
13144 KB |
Output is correct |
9 |
Correct |
2 ms |
13144 KB |
Output is correct |
10 |
Incorrect |
3 ms |
13144 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '83005428' |
11 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
415 ms |
14932 KB |
1st lines differ - on the 1st token, expected: '431985922', found: '646726998' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
415 ms |
14932 KB |
1st lines differ - on the 1st token, expected: '431985922', found: '646726998' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
13144 KB |
Output is correct |
2 |
Incorrect |
3 ms |
13144 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '83005428' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
13144 KB |
Output is correct |
2 |
Correct |
2 ms |
13144 KB |
Output is correct |
3 |
Correct |
2 ms |
13144 KB |
Output is correct |
4 |
Correct |
3 ms |
13144 KB |
Output is correct |
5 |
Correct |
3 ms |
13144 KB |
Output is correct |
6 |
Correct |
3 ms |
13140 KB |
Output is correct |
7 |
Correct |
2 ms |
13144 KB |
Output is correct |
8 |
Correct |
2 ms |
13144 KB |
Output is correct |
9 |
Correct |
2 ms |
13144 KB |
Output is correct |
10 |
Incorrect |
3 ms |
13144 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '83005428' |
11 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
13144 KB |
Output is correct |
2 |
Correct |
2 ms |
13144 KB |
Output is correct |
3 |
Correct |
2 ms |
13144 KB |
Output is correct |
4 |
Correct |
3 ms |
13144 KB |
Output is correct |
5 |
Correct |
3 ms |
13144 KB |
Output is correct |
6 |
Correct |
3 ms |
13140 KB |
Output is correct |
7 |
Correct |
2 ms |
13144 KB |
Output is correct |
8 |
Correct |
2 ms |
13144 KB |
Output is correct |
9 |
Correct |
2 ms |
13144 KB |
Output is correct |
10 |
Incorrect |
3 ms |
13144 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '83005428' |
11 |
Halted |
0 ms |
0 KB |
- |