# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
790453 | borisAngelov | Digital Circuit (IOI22_circuit) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "circuit.h"
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1005;
const long long mod = 1e9 + 2022;
int n, m;
long long dp[maxn][2];
int color[maxn];
vector<int> tree[2 * maxn];
void dfs(int node)
{
if (tree[node].empty())
{
dp[node][color[node]] = 1;
dp[node][1 - color[node]] = 0;
return;
}
int l = tree[node][0];
int r = tree[node][1];
dfs(l);
dfs(r);
dp[node][0] = dp[l][0] * dp[r][1] + dp[l][1] * dp[r][0] + 2 * dp[l][0] * dp[r][0];
dp[node][0] %= mod;
dp[node][1] = dp[l][0] * dp[r][1] + dp[l][1] * dp[r][0] + 2 * dp[l][1] * dp[r][1];
dp[node][1] %= mod;
}
void init(int N, int M, vector<int> P, vector<int> A)
{
n = N;
m = M;
for (int i = 1; i < p.size(); ++i)
{
tree[P[i] + 1].push_back(i + 1);
}
for (int i = n + 1; i <= n + m; ++i)
{
color[i] = A[n - i - 1];
}
}
int count_ways(int L, int R)
{
int l = ++L;
int r = ++R;
for (int i = l; i <= r; ++i)
{
color[i] = 1 - color[i];
}
dfs(1);
return dp[1][1];
}