#include "circuit.h"
#include<bits/stdc++.h>
#include <vector>
using namespace std;
#define pb push_back
#define fi first
#define se second
#define ll long long
const int N=1e5+50;
const ll mod=1000002022;
ll Puta(ll a,ll b){a%=mod,b%=mod;ll x=a*b;x%=mod;return x;}
ll Plus(ll a,ll b){a%=mod,b%=mod;ll x=a+b;x%=mod;return x;}
vector<int>E[N];
int a[N],n,m;
ll res[N],dp[N][2],F[N];
/*void DFSsetup(int u){
F[u]=E[u].size()+1;
for(auto i:E[u]){
DFSsetup(i);
F[u]*=F[i];
}
}*/
void DFS(int u){
if(E[u].empty()) return;
for(auto i:E[u]){
DFS(i);
}
dp[u][1]=dp[u][0]=0;
int mask=0;
while(mask< 1<<(E[u].size())){
ll temp1=1,sum=0;
for(int i=0,temp=mask;i<E[u].size();i++,temp/=2){
temp1=Puta(temp1,dp[E[u][i]][temp%2]);sum+=temp%2;
}
if(mask!=0) dp[u][1]=Plus(dp[u][1],Puta(sum,temp1));
dp[u][0]=Plus(dp[u][0],Puta(E[u].size()-sum,temp1));
mask++;
}
//dp[u][0]=F[u]-dp[u][1];
}
void init(int N1, int M1, std::vector<int> P, std::vector<int> A) {
n=N1,m=M1;
for(int i=0;i<P.size();i++){
E[P[i]].pb(i);
}
for(int i=0;i<m;i++){
dp[n+i][1]=A[i],dp[n+i][0]=1-A[i];
a[i]=A[i];
}
//DFSsetup(0);
DFS(0);
/*for(int i=0;i<=n+m-1;i++){
printf("%i: %i %i\n",i,dp[i][0],dp[i][1]);
}*/
/*for(int i=1;i<=M;i++){
}*/
}
int count_ways(int L, int R) {
if(n==1){
int ct;
for(int i=L;i<=R;i++) a[i]=1-a[i];
for(int i=0;i<=n+m-1;i++) ct+=a[i];
return ct;
}
for(int i=L;i<=R;i++){
dp[i][1]=1-dp[i][1];
dp[i][0]=1-dp[i][0];
}
DFS(0);
/*for(int i=0;i<=n+m-1;i++){
printf("%i: %i %i\n",i,dp[i][0],dp[i][1]);
}*/
return dp[0][1];
}
Compilation message
circuit.cpp: In function 'void DFS(int)':
circuit.cpp:32:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
32 | for(int i=0,temp=mask;i<E[u].size();i++,temp/=2){
| ~^~~~~~~~~~~~
circuit.cpp: In function 'void init(int, int, std::vector<int>, std::vector<int>)':
circuit.cpp:43:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
43 | for(int i=0;i<P.size();i++){
| ~^~~~~~~~~
circuit.cpp: In function 'int count_ways(int, int)':
circuit.cpp:62:7: warning: 'ct' may be used uninitialized in this function [-Wmaybe-uninitialized]
62 | int ct;
| ^~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
1 ms |
5976 KB |
1st lines differ - on the 1st token, expected: '1', found: '11' |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
2 ms |
5976 KB |
1st lines differ - on the 1st token, expected: '1', found: '11' |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
1 ms |
5976 KB |
1st lines differ - on the 1st token, expected: '1', found: '11' |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
3069 ms |
7808 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
3069 ms |
7808 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
2 ms |
5976 KB |
1st lines differ - on the 1st token, expected: '1', found: '11' |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
1 ms |
5976 KB |
1st lines differ - on the 1st token, expected: '1', found: '11' |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
1 ms |
5976 KB |
1st lines differ - on the 1st token, expected: '1', found: '11' |
| 2 |
Halted |
0 ms |
0 KB |
- |
