Submission #1219801

#TimeUsernameProblemLanguageResultExecution timeMemory
1219801bangan디지털 회로 (IOI22_circuit)C++20
0 / 100
1040 ms2162688 KiB
#include "circuit.h"

// #include <vector>
#include <bits/stdc++.h>

using i64 = long long;

constexpr int mod = 1'000'002'022;

int n, m;
std::vector<int> p, a;
std::vector<std::vector<int>> inp;

std::vector<std::array<int, 2>> f;
std::vector<std::vector<int>> g;

void init(int N, int M, std::vector<int> P, std::vector<int> A) {
  n = N;
  m = M;
  p = P;
  a = A;

  inp.resize(n);
  for (int i = 1; i < n + m; i++) {
    inp[p[i]].push_back(i);
  }

  for (int i = 0; i < n; i++) {
    assert(inp[i].size() < n + m);
  }

  f.resize(n + m);
  g.resize(n);
  for (int i = 0; i < n; i++) {
    g[i].resize(n + m);
  }
}

void clear() {
  for (int i = 0; i < n + m; i++) {
    f[i] = {0, 0};
  }
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < n + m; j++) {
      g[i][j] = 0;
    }
  }
}

int count_ways(int L, int R) {
  for (int i = L - n; i <= R - n; i++) {
    a[i] ^= 1;
  }

  clear();
  for (int i = n; i < n + m; i++) {
    f[i][a[i - n]] = 1;
  }

  for (int i = n - 1; i >= 0; i--) {
    g[i][0] = 1;
    for (int j : inp[i]) {
      std::vector<int> new_g(n + m);
      new_g[0] = i64(g[i][0]) * f[j][0] % mod;

      for (int k = 1; k < n + m; k++) {
        new_g[k] = (i64(g[i][k]) * f[j][0] % mod
          + i64(g[i][k - 1]) * f[j][1] % mod) % mod;
      }
      g[i] = new_g;
    }

    int less = g[i][0];
    int more = std::accumulate(g[i].begin() + 1, g[i].end(), 0ll) % mod;
    for (int t = 1; t <= inp.size(); i++) {
      f[i][0] = (f[i][0] + less) % mod;
      f[i][1] = (f[i][1] + more) % mod;

      less += g[i][t];
      more -= g[i][t];
    }
  }

  return f[0][1];
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...