#include "circuit.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef vector<vi> vii;
typedef pair<int,int> pi;
typedef vector<pi> pii;
typedef tuple<ll,int,int> ti;
typedef vector<ll> li;
typedef vector<li> lii;
#define REP(i,a,b) for(int i=a;i<b;i++)
#define F first
#define S second
#define PB push_back
#define LSOne(s) ((s)&(-s))
#define all(x) (x).begin(),(x).end()
ll INF=1000000000000000010;
int inf=1e9+10;
ll M=1000002022;
int n,m,k;
vi a,p,lazy;
vector<ll> X,Y,Z,ch;
vector<vector<ll>> pref,suff;
vii ed;
void dfs(int x,ll y){
if(ed[x].empty()){
Z[x-n]=y;
return;
}
REP(i,0,ed[x].size()){
ll q=1,r=1;
if(i!=0)q=pref[x][i-1];
if(i!=ed[x].size()-1)r=suff[x][i+1];
dfs(ed[x][i],(((y*q)%M)*r)%M);
}
}
void build(int x,int l,int r){
if(l==r){
if(a[l])X[x]=Z[l];
else Y[x]=Z[l];
return;
}
int mid=(l+r)/2;
build(2*x,l,mid);
build(2*x+1,mid+1,r);
X[x]=(X[2*x]+X[2*x+1])%M;
Y[x]=(Y[2*x]+Y[2*x+1])%M;
}
void update(int x,int l,int r,int u,int v){
if(u>v)return;
if(lazy[x]%2){
lazy[x]=0;
if(l!=r){
swap(X[2*x],Y[2*x]);
swap(X[2*x+1],Y[2*x+1]);
lazy[2*x]++;
lazy[2*x+1]++;
}
}
if(l==u&&r==v){
swap(X[x],Y[x]);
lazy[x]++;
return;
}
int mid=(l+r)/2;
update(2*x,l,mid,u,min(v,mid));
update(2*x+1,mid+1,r,max(u,mid+1),v);
X[x]=(X[2*x]+X[2*x+1])%M;
Y[x]=(Y[2*x]+Y[2*x+1])%M;
}
void init(int N, int MM, std::vector<int> P, std::vector<int> A) {
n=N;m=MM;p=P;a=A;
k=n+m;
X.resize(4*m+1,0);
Y.resize(4*m+1,0);
Z.resize(m,0);
lazy.resize(4*m+1,0);
ed.resize(n+m);
ch.resize(n+m,0);
REP(i,1,n+m)ed[p[i]].PB(i);
REP(i,1,n+m)ch[p[i]]++;
REP(i,n,n+m)ch[i]=1;
for(int i=n-1;i>=1;i--){
ch[p[i]]*=ch[i];
ch[p[i]]%=M;
}
pref.resize(n);suff.resize(n);
REP(i,0,n){
int sz=ed[i].size();
pref[i].resize(sz);
suff[i].resize(sz);
REP(j,0,sz){
pref[i][j]=ch[ed[i][j]];
suff[i][j]=ch[ed[i][j]];
}
REP(j,1,sz){
pref[i][j]*=pref[i][j-1];
pref[i][j]%=M;
}
for(int j=sz-2;j>=0;j--){
suff[i][j]*=suff[i][j+1];
suff[i][j]%=M;
}
}
dfs(0,1LL);
build(1,0,m-1);
return;
}
int count_ways(int L, int R) {
update(1,0,m-1,L-n,R-n);
return X[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... |