Submission #1074124

#	Time	Username	Problem	Language	Result	Execution time	Memory
1074124		fv3	Digital Circuit (IOI22_circuit)	C++17	7 / 100	3076 ms	3928 KiB

circuit

#include "circuit.h"
#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

const ll MOD = 1000002022ll;

int N, M;
vector<int> A;

vector<vector<int>> adj;
vector<pair<ll, ll>> nodes;

void DFS(int index)
{
  if (index >= N)
  {
    if (A[index - N])
      nodes[index] = {1, 0};
    else
      nodes[index] = {0, 1};
    return;
  }

  for (auto node : adj[index])
  {
    DFS(node);
  }

  const auto l = nodes[adj[index][0]];
  const auto r = nodes[adj[index][1]];

  nodes[index] = 
  { 
   (2 * l.first * r.first + 
        l.first * r.second + 
        l.second * r.first ) % MOD,
   (2 * l.second * r.second + 
        l.first * r.second + 
        l.second * r.first ) % MOD
  };
}

void init(int N_, int M_, vector<int> P, vector<int> A_) 
{
  N = N_;
  M = M_;
  A = A_;

  adj = vector<vector<int>>(N);
  for (int i = 1; i < N + M; i++)
    adj[P[i]].push_back(i);

  nodes = vector<pair<ll, ll>>(N+M);
}

int count_ways(int L, int R) 
{
  L -= N; R -= N;

  for (int i = L; i <= R; i++)
    A[i] ^= 1;

  DFS(0);
  return nodes[0].first;
}

• 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