Submission #837515

#	Time	Username	Problem	Language	Result	Execution time	Memory
837515		fatemetmhr	Digital Circuit (IOI22_circuit)	C++17	2 / 100	23 ms	19568 KiB

circuit

// Be name khoda //


#include "circuit.h"
#include <bits/stdc++.h>

using namespace std;

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

#define debug(x) cerr << "(" << (#x) << "): " << (x) << endl;
#define all(x)   x.begin(), x.end()
#define pb       push_back
#define mp       make_pair
#define fi       first
#define se       second

typedef long long ll;

const int mod   = 1000002022;
const int maxn5 = 1e3 + 10;

int dp[maxn5][2], cnt[maxn5][2][maxn5], ty[maxn5];
vector <int> adj[maxn5];


void dfs(int v){
    if(adj[v].empty()){
        dp[v][1] = ty[v];
        dp[v][0] = ty[v] ^ 1;
        return; 
    }
    cnt[v][1][0] = 1;
    int x = 0;
    for(auto u : adj[v]){
        dfs(u);
        for(int i = 0; i <= int(adj[v].size()); i++)
            cnt[v][x][i] = (cnt[v][x ^ 1][i] * ll(dp[u][0]) % mod + (i ? cnt[v][x ^ 1][i - 1] * ll(dp[u][1]) % mod : 0)) % mod;
        x ^= 1;
    }
    for(int i = 0; i <= int(adj[v].size()); i++){
        dp[v][0] = (dp[v][0] + cnt[v][x ^ 1][i] * ll(int(adj[v].size()) - i)) % mod;
        dp[v][1] = (dp[v][1] + cnt[v][x ^ 1][i] * ll(i)) % mod;
    }
}

void init(int n, int m, std::vector<int> p, std::vector<int> a) {
    for(int i = 1; i < n + m; i++)
        adj[p[i]].pb(i);
    for(int i = 0; i < m; i++)
        ty[i + n] = a[i];
}

int count_ways(int l, int r) {
    memset(dp, 0, sizeof dp);
    memset(cnt, 0, sizeof cnt);
    for(int i = l; i <= r; i++)
        ty[i] ^= 1;
    dfs(0);
    return dp[0][1];
}

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