Submission #826180

#	Time	Username	Problem	Language	Result	Execution time	Memory
826180		LittleCube	Digital Circuit (IOI22_circuit)	C++17	6 / 100	3078 ms	11668 KiB

circuit

#include "circuit.h"
#include <bits/stdc++.h>
#define ll long long
using namespace std;

const ll mod = 1'000'002'022;
namespace
{
  int N, M;
  vector<int> P, A, c;
  vector<int> child[200005];
  ll zero[200005], one[200005];
}

void dothings(int i)
{
  int K = child[i].size();
  vector<ll> dp(K + 1, 0);
  dp[0] = 1;
  for (auto j : child[i])
    for (int k = K - 1; k >= 0; k--)
    {
      dp[k + 1] = (dp[k + 1] + dp[k] * one[j]) % mod;
      dp[k] = (dp[k] * zero[j]) % mod;
    }
  zero[i] = one[i] = 0;
  for (int k = 0; k <= K; k++)
  {
    zero[i] = (zero[i] + (K - k) * dp[k]) % mod;
    one[i] = (one[i] + k * dp[k]) % mod;
  }
}

void init(int N, int M, vector<int> P, vector<int> A)
{
  ::N = N;
  ::M = M;
  ::P = P;
  ::A = A;
  c.resize(N);
  for (int i = 1; i < N + M; i++)
    c[P[i]]++, child[P[i]].emplace_back(i);
 
  for (int i = 0; i < M; i++)
    if (A[i] == 1)
      zero[N + i] = 0, one[N + i] = 1;
    else
      zero[N + i] = 1, one[N + i] = 0;
  for (int i = N - 1; i >= 0; i--)
    dothings(i);
}

int count_ways(int L, int R)
{
  L -= N, R -= N;
  for (int i = L; i <= R; i++)
    A[i] ^= 1;
  for (int i = L; i <= R; i++)
    if (A[i] == 1)
      zero[N + i] = 0, one[N + i] = 1;
    else
      zero[N + i] = 1, one[N + i] = 0;
  for (int i = P[L + N]; i >= 0; i = P[i])
    dothings(i);
  return one[0];
}

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