답안 #922942

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
922942 2024-02-06T10:22:29 Z vjudge1 디지털 회로 (IOI22_circuit) C++17
컴파일 오류
0 ms 0 KB
#include <bits/stdc++.h>
#define pb push_back

using namespace std;

const int maxN = 2e5 + 5;
const int mod = 1000002022;

void add(int &a, int b){
    a += b;
    if (a >= mod) a -= mod;
    if (a < 0) a += mod;
}

vector <int> adj[maxN], pre[maxN], suf[maxN];
int sz[maxN], val[maxN];

void dfs(int u){
    int dmm = adj[u].size();
    sz[u] = 1;
    for (int i = 0; i < dmm; ++ i){
        int v = adj[u][i];
        dfs(v);
        if (!i) pre[u].pb(sz[v]);
        else pre[u].pb(1ll * pre[u].back() * sz[v] % mod);
        suf[u].pb(0);
        sz[u] += sz[v];
    }
    for (int i = dmm - 1; i >= 0; -- i){
        int v = adj[u][i];
        if (i == int(dmm - 1)) suf[u][i] = sz[v];
        else suf[u][i] = 1ll * suf[u][i + 1] * sz[v] % mod;
    }
}

void get(int u){
    int dmm = adj[u].size();
    for (int i = 0; i < dmm; ++ i){
        int v = adj[u][i];
        val[v] = val[u];
        if (i > 0) val[v] = 1ll * val[v] * pre[u][i - 1] % mod;
        if (i < dmm - 1) val[v] = 1ll * val[v] * suf[u][i + 1] % mod;
    }
}

int sum[2][4 * maxN];
bool used[maxN], rev[maxN];

void mer(int s){
    sum[0][s] = 0;
    sum[1][s] = 0;
    for (int i = 0; i <= 1; ++ i){
        for (int j = 2 * s; j <= 2 * s + 1; ++ j) add(sum[i][s], sum[i][j]);
    }
}

void build(int s, int l, int r, int n){
    rev[s] = 0;
    if (l == r){
        sum[0][s] = used[l] * val[l + n];
        sum[1][s] = (1 ^ used[l]) * val[l + n];
        return;
    }

    int mid = (l + r) / 2;
    build(2 * s, l, mid, n);
    build(2 * s + 1, mid + 1, r, n);
    mer(s);
}

int kk, gg;

void init(int N, int M, int[] P, int[] A){
    kk = N;
    gg = m;
    for (int i = 1; i < N + m; ++ i) adj[P[i]].pb(i);
    dfs(0);
    val[0] = 1;
    get(0);
    for (int i = 0; i < m; ++ i) used[i] = A[i];
    build(1, 0, m - 1, N);
}

void down(int s){
    if (!rev[s]) return;
    for (int i = 2 * s; i <= 2 * s + 1; ++ i) {
        rev[i] ^= 1;
        swap(sum[0][i], sum[1][i]);
    }
    rev[s] = 0;
}

void upd(int s, int l, int r, int u, int v){
    if (u <= l && r <= v){
        rev[s] ^= 1;
        swap(sum[0][s], sum[1][s]);
        return;
    }

    int mid = (l + r) / 2;
    down(s);
    if (mid >= u) upd(2 * s, l, mid, u, v);
    if (mid + 1 <= v) upd(2 * s + 1, mid + 1, r, u, v);
    mer(s);
}

int count_ways(int L, int R){
    upd(1, 0, gg - 1, L - kk, R - kk);
    return sum[0][1];
}

Compilation message

circuit.cpp:73:31: error: expected ',' or '...' before 'P'
   73 | void init(int N, int M, int[] P, int[] A){
      |                               ^
circuit.cpp: In function 'void init(int, int, int*)':
circuit.cpp:75:10: error: 'm' was not declared in this scope
   75 |     gg = m;
      |          ^
circuit.cpp:76:42: error: 'P' was not declared in this scope
   76 |     for (int i = 1; i < N + m; ++ i) adj[P[i]].pb(i);
      |                                          ^
circuit.cpp:80:44: error: 'A' was not declared in this scope
   80 |     for (int i = 0; i < m; ++ i) used[i] = A[i];
      |                                            ^