oj.uz

Submission #866785

# Submission time Handle Problem Language Result Execution time Memory
866785 2023-10-27T04:10:09 Z Pety Digital Circuit (IOI22_circuit) C++17
2 / 100
 378 ms 10844 KB 
#include <bits/stdc++.h>


using namespace std;

const int mod = 1000002022;
const int N = 1e5+2;
const int NM = 2e5+2;
int n, m, val[NM], lazy[4*N], contrib[N];
vector<int>G[NM];

pair<int, int>aint[4*N];

pair<int, int>init_contrib[NM];

void dfs (int nod) {
  if (nod >= n) {
    val[nod] = 1;
    return;
  }
  val[nod] = G[nod].size();
  for (auto it : G[nod]) {
    dfs(it);
    val[nod] = 1ll * val[it] * val[nod] % mod;
  }
} 

void dfs_contrib (int nod, int c) {
  if (nod >= n) {
    contrib[nod - n] = c;
    return;
  }
  vector<int>pref(G[nod].size());
  vector<int>suf(G[nod].size());
  for (int i = 0; i < G[nod].size(); i++) {
    int child = G[nod][i];
    pref[i] = val[child];
    if (i)
      pref[i] = 1ll * pref[i - 1] * pref[i] % mod;
  }
  for (int i = G[nod].size() - 1; i >= 0; i--) {
    int child = G[nod][i];
    suf[i] = val[child];
    if (i + 1 < G[nod].size())
      suf[i] = 1ll * suf[i] * suf[i + 1] % mod;
  }
  for (int i = 0; i < G[nod].size(); i++) {
    int child = G[nod][i];
    int val = c;
    if (i > 0)
      val = 1ll * val * pref[i - 1] % mod;
    if (i + 1 < G[nod].size())
      val = 1ll * val * suf[i + 1] % mod;
    dfs_contrib(child, val);
  }
}

void build (int nod, int st, int dr) {
  if (st == dr) {
    aint[nod] = init_contrib[st];
    return;
  }
  int mid = (st + dr) / 2;
  build(2 * nod, st, mid);
  build(2 * nod + 1, mid + 1, dr);
  aint[nod].first = aint[2 * nod].first + aint[2 * nod + 1].first;
  aint[nod].second = aint[2 * nod].second + aint[2 * nod + 1].second;
}

void push (int nod) {
  if (!lazy[nod])
    return;
  for (int i = 0; i < 2; i++) {
    swap(aint[2 * nod + i].first, aint[2 * nod + i].second);
    lazy[2 * nod + i] ^= 1;
  }
  lazy[nod] = 0;
}

void update (int nod, int st, int dr, int a, int b) {
  if (a <= st && dr <= b) {
    swap(aint[nod].first, aint[nod].second);
    lazy[nod] ^= 1;
    return;
  }
  int mid = (st + dr) / 2;
  push(nod);
  if (a <= mid)
    update(2 * nod, st, mid, a, b);
  if (b > mid)
    update(2 * nod + 1, mid + 1, dr, a, b);
  aint[nod].first = aint[2 * nod].first + aint[2 * nod + 1].first;
  aint[nod].second = aint[2 * nod].second + aint[2 * nod + 1].second;
}


void init (int _n, int _m, vector<int>P, vector<int> A) {
  n = _n;
  m = _m;
  for (int i = 1; i < P.size(); i++)
    G[P[i]].push_back(i);
  dfs(0);
  dfs_contrib(0, 1);
  for (int i = 0; i < m; i++) {
    pair<int, int>c = {0, contrib[i]};
    if (!A[i])
      swap(c.first, c.second);
    init_contrib[i] = c;
  }
  build(1, 0, m - 1);
}

int count_ways(int l, int r) {
  update(1, 0, m - 1, l - n, r - n);
  return aint[1].second;
}

Compilation message

circuit.cpp: In function 'void dfs_contrib(int, int)':
circuit.cpp:35:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   35 |   for (int i = 0; i < G[nod].size(); i++) {
      |                   ~~^~~~~~~~~~~~~~~
circuit.cpp:44:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   44 |     if (i + 1 < G[nod].size())
      |         ~~~~~~^~~~~~~~~~~~~~~
circuit.cpp:47:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   47 |   for (int i = 0; i < G[nod].size(); i++) {
      |                   ~~^~~~~~~~~~~~~~~
circuit.cpp:52:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   52 |     if (i + 1 < G[nod].size())
      |         ~~~~~~^~~~~~~~~~~~~~~
circuit.cpp: In function 'void init(int, int, std::vector<int>, std::vector<int>)':
circuit.cpp:100:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  100 |   for (int i = 1; i < P.size(); i++)
      |                   ~~^~~~~~~~~~

Subtask #1
2.0 / 2.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 8536 KB Output is correct
2 Correct 1 ms 8536 KB Output is correct
3 Correct 1 ms 8536 KB Output is correct
4 Correct 1 ms 8536 KB Output is correct
5 Correct 2 ms 8536 KB Output is correct
6 Correct 2 ms 8536 KB Output is correct
7 Correct 2 ms 8536 KB Output is correct
8 Correct 2 ms 8536 KB Output is correct

Subtask #2
0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 8536 KB Output is correct
2 Incorrect 1 ms 8536 KB 1st lines differ - on the 1st token, expected: '52130940', found: '-1782334720'
3 Halted 0 ms 0 KB -

Subtask #3
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 8536 KB Output is correct
2 Correct 1 ms 8536 KB Output is correct
3 Correct 1 ms 8536 KB Output is correct
4 Correct 1 ms 8536 KB Output is correct
5 Correct 2 ms 8536 KB Output is correct
6 Correct 2 ms 8536 KB Output is correct
7 Correct 2 ms 8536 KB Output is correct
8 Correct 2 ms 8536 KB Output is correct
9 Correct 1 ms 8536 KB Output is correct
10 Incorrect 1 ms 8536 KB 1st lines differ - on the 1st token, expected: '52130940', found: '-1782334720'
11 Halted 0 ms 0 KB -

Subtask #4
0 / 4.0

# Verdict Execution time Memory Grader output
1 Incorrect 378 ms 10844 KB 1st lines differ - on the 1st token, expected: '431985922', found: '-747097920'
2 Halted 0 ms 0 KB -

Subtask #5
0 / 12.0

# Verdict Execution time Memory Grader output
1 Incorrect 378 ms 10844 KB 1st lines differ - on the 1st token, expected: '431985922', found: '-747097920'
2 Halted 0 ms 0 KB -

Subtask #6
0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 8536 KB Output is correct
2 Incorrect 1 ms 8536 KB 1st lines differ - on the 1st token, expected: '52130940', found: '-1782334720'
3 Halted 0 ms 0 KB -

Subtask #7
0 / 28.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 8536 KB Output is correct
2 Correct 1 ms 8536 KB Output is correct
3 Correct 1 ms 8536 KB Output is correct
4 Correct 1 ms 8536 KB Output is correct
5 Correct 2 ms 8536 KB Output is correct
6 Correct 2 ms 8536 KB Output is correct
7 Correct 2 ms 8536 KB Output is correct
8 Correct 2 ms 8536 KB Output is correct
9 Correct 1 ms 8536 KB Output is correct
10 Incorrect 1 ms 8536 KB 1st lines differ - on the 1st token, expected: '52130940', found: '-1782334720'
11 Halted 0 ms 0 KB -

Subtask #8
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 8536 KB Output is correct
2 Correct 1 ms 8536 KB Output is correct
3 Correct 1 ms 8536 KB Output is correct
4 Correct 1 ms 8536 KB Output is correct
5 Correct 2 ms 8536 KB Output is correct
6 Correct 2 ms 8536 KB Output is correct
7 Correct 2 ms 8536 KB Output is correct
8 Correct 2 ms 8536 KB Output is correct
9 Correct 1 ms 8536 KB Output is correct
10 Incorrect 1 ms 8536 KB 1st lines differ - on the 1st token, expected: '52130940', found: '-1782334720'
11 Halted 0 ms 0 KB -