Submission #793048

#TimeUsernameProblemLanguageResultExecution timeMemory
793048SlavicG디지털 회로 (IOI22_circuit)C++17
0 / 100
3083 ms24272 KiB
#include <bits/stdc++.h>
#include "circuit.h"
using namespace std;
const int N = 5e5 + 10, mod = 1e9 + 2;
vector<int> adj[N];
int64_t dp[N][2], color[N], in[N], s[N], tt = 0;
void dfs(int u) {
    s[u] = 1;
    in[u] = tt++;
    assert(adj[u].size() == 2 || adj[u].size() == 0);
    if(adj[u].size() == 0) {
        if(color[u]) dp[u][0] = 0, dp[u][1] = 1;
        else dp[u][0] = 1, dp[u][1] = 0;
        return;
    }
    int v1 = adj[u][0], v2 = adj[u][1];
    dfs(v1);
    dfs(v2);
    s[u] += s[v1];
    s[u] += s[v2];
    dp[u][1] = dp[v1][1] * dp[v2][1] % mod * 2 % mod + (dp[v1][0] * dp[v2][1]) % mod + (dp[v1][1] * dp[v2][0]) % mod;
    dp[u][1] %= mod;
    dp[u][0] = dp[v1][0] * dp[v2][0] % mod * 2 % mod + (dp[v1][0] * dp[v2][1]) % mod + (dp[v1][1] * dp[v2][0]) % mod;
    dp[u][0] %= 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]].push_back(i);
    }
    for(int i = 0; i < M; ++i) color[i + N] = A[i];
}

int count_ways(int L, int R) {
    for(int i = L; i <= R; ++i) {
        color[i] ^= 1;
    }
    dfs(0);
    return dp[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...