Submission #1053606

#	Time	Username	Problem	Language	Result	Execution time	Memory
1053606		Ignut	Digital Circuit (IOI22_circuit)	C++17	18 / 100	3054 ms	7256 KiB

circuit

/* Ignut
started: 11.08.2024
now: 11.08.2024
████████████████████████████████████████████████████████████████████
████████████████████████████████    ████████████████████████████████
██████████████████████████████        ██████████████████████████████
██████      ██████████████████        ██████████████████      ██████
██████          ██████████████        ██████████████          ██████
██████      ██    ████████████        ████████████    ██      ██████
██████      ████    ██████████        ██████████    ████      ██████
██████      ████      ██████████    ██████████      ████      ██████
██████      ████      ██████████    ██████████    ██████      ██████
██████      ██████    ██████████    ██████████    ██████      ██████
██████      ██████    ████████        ████████    ██████      ██████
██████      ██████      ██████        ██████      ██████      ██████
██████      ████        ████            ████        ████      ██████
██████            ██████████    ████    ██████████            ██████
██████      ██      ██████    ████████    ██████      ██      ██████
██████      ██████            ████████            ██████      ██████
██████                    ██            ██                    ██████
██████████████████████      ████    ████      ██████████████████████
████████████████████████      ██    ██      ████████████████████████
██████████████████████████                ██████████████████████████
██████████████████████████████        ██████████████████████████████
████████████████████████████████████████████████████████████████████
*/

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

const int MOD = 1e9 + 2022;

void vadd(int &a, int b) {
    a += a + b >= MOD ? b - MOD : b;
}

int mult(int a, int b) {
    return 1ll * a * b % MOD;
}

// ===================================================== //

const int MAXN = 2e5 + 123;
int n, m;

vector<int> p, a;
vector<int> tree[MAXN];

void init(int N, int M, vector<int> P, vector<int> A) {
    n = N, m = M;
    a = A, p = P;
    for (int i = 0; i < n + m; i ++) {
        if (p[i] == -1) continue;
        tree[p[i]].push_back(i);
    }
}

pair<int, int> dfs(int v) {
    if (v >= n) {
        // cout << v << " : " << (1 ^ a[v - n]) << ", " << a[v - n] << '\n';
        return {1 ^ a[v - n], a[v - n]};
    }

    vector<pair<int, int>> w;
    for (int to : tree[v]) {
        w.push_back(dfs(to));
    }
    int sz = w.size();
    int dp[sz + 1] = {};
    dp[0] = 1;
    for (auto [prob0, prob1] : w) {
        vector<int> nxt(sz + 1, 0);
        for (int i = sz - 1; i >= 0; i --) {
            vadd(nxt[i + 1], mult(dp[i], prob1));
            vadd(nxt[i], mult(dp[i], prob0));
        }
        for (int i = 0; i <= sz; i ++) dp[i] = nxt[i];
    }
    // cout << v << " -- " << sz << '\n';
    // for (int i = 0; i <= sz; i ++) cout << dp[i] << ' ';
    // cout << '\n';
    int r0 = 0, r1 = 0;
    for (int i = 0; i <= sz; i ++) {
        vadd(r1, mult(dp[i], i));
        vadd(r0, mult(dp[i], sz - i));
    }
    // cout << v << " : " << r0 << ", " << r1 << '\n';
    return {r0, r1};
}

int count_ways(int L, int R) {
    L -= n, R -= n;
    for (int i = L; i <= R; i ++) a[i] ^= 1;
    
    int res = (dfs(0)).second;
    return res;
}

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