Submission #869725

#	Time	Username	Problem	Language	Result	Execution time	Memory
869725		Ludissey	디지털 회로 (IOI22_circuit)	C++17	7 / 100	3016 ms	2097152 KiB

circuit

#include "circuit.h"

#include <bits/stdc++.h>

using namespace std;
#define int long long

int N,M;
const int MOD=1e9+2022;
vector<vector<int>> tree;
vector<int> state;
vector<int> depth;
vector<int> poss;
vector<vector<int>> memo;

int dfs(int i, int on){
  if(i>=N){
    if(on) return state[i];
    else return 1-state[i];
  }
  if(memo[i][on]!=-1) return memo[i][on];
  int lo=dfs(tree[i][0],true)%MOD, lf=dfs(tree[i][0],false)%MOD;
  int ro=dfs(tree[i][1],true)%MOD, rf=dfs(tree[i][1],false)%MOD;
  int ret=(lo*rf)%MOD+(ro*lf)%MOD;
  if(on) ret+=(lo*ro*2)%MOD;
  else ret+=(2*lf*rf)%MOD;
  return memo[i][on]=ret%MOD;
}

void init(signed n, signed m, std::vector<signed> P, std::vector<signed> A) {
  N=n; M=m;
  tree.resize(N+M); state.resize(N+M);
  for (int i = 1; i < N+M; i++) tree[P[i]].push_back(i);
  for (int i = N; i < N+M; i++) state[i] = A[i-N];
}

signed count_ways(signed L, signed R) {
  poss.clear(); poss.resize(N+M,0);
  memo.clear(); memo.resize(N+M, vector<int>(2,-1));
  for (int i = L; i <= R; i++) state[i] = 1-state[i];
  signed d=dfs(0,true)%MOD;
  return d;
}

• Subtask 1: 0 / 2
• Subtask 2: 7 / 7
• Subtask 3: 0 / 9
• Subtask 4: 0 / 4
• Subtask 5: 0 / 12
• Subtask 6: 0 / 27
• Subtask 7: 0 / 28
• Subtask 8: 0 / 11