#include "circuit.h"
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
using namespace std;
const long long INF = 1e17;
typedef long long ll;
const ll MOD = 1000002022;
#define S second
#define F first
#define pb push_back
#define V vector
#define all(v) v.begin(), v.end()
V<int>adj[(int)1e5+1];
ll dp[(int)2e5+1];
ll dp1[(int)2e5+1];
int a[(int)2e5+1];
ll b[(int)2e5+1];
int na,ma;
ll add(ll a,ll b) {
return ((a%MOD)+(b%MOD))%MOD;
}
ll mul(ll a,ll b) {
return ((a%MOD*(b%MOD)))%MOD;
}
void dfs(int n,int p) {
ll ch=(ll)adj[n].size()-1+(n==0);
if (ch==0) {
dp[n]=1;
return;
}
dp[n]=ch;
for (auto u:adj[n]) {
if (u==p)continue;
dfs(u,n);
dp[n]=mul(dp[n],dp[u]);
}
}
void dfs1(int n,int p) {
ll ch=(ll)adj[n].size()-1+(n==0);
if (ch==0) {
return;
}
V<ll>pref(ch);
V<ll>suff(ch);
int c=0;
for (auto u:adj[n]) {
if (u==p)continue;
pref[c]=dp[u];
suff[c]=dp[u];
c++;
}
for (int i=1;i<c;i++) {
pref[i]=mul(pref[i-1],pref[i]);
}
for (int i=c-2;i>=0;i--) {
suff[i]=mul(suff[i+1],suff[i]);
}
c=0;
for (auto u:adj[n]) {
if (u==p)continue;
ll x=1;
if (c>0) {
x=mul(x,pref[c-1]);
}
if (c<ch-1) {
x=mul(x,suff[c+1]);
}
dp1[u]=mul(x,dp1[n]);
c++;
}
for (auto u:adj[n]) {
if (u==p)continue;
dfs1(u,n);
}
}
void init(int N, int M, std::vector<int> P, std::vector<int> A) {
na=N;
ma=M;
for (int i =1;i<N+M;i++) {
adj[i].pb(P[i]);
adj[P[i]].pb(i);
}
dp1[0]=1;
dfs(0,-1);
dfs1(0,-1);
for (int i=0;i<M;i++) {
a[i+N]=A[i];
b[i+N]=dp1[i+N];
}
}
int count_ways(int L, int R) {
for (int i=L;i<=R;i++) {
a[i]^=1;
}
ll x=0;
for (int i=0;i<ma;i++) {
if (a[i+na]==1) {
x=add(x,b[i+na]);
}
}
x%=MOD;
return (int)x;
}
# | 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... |