# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
627373 | Elegia | 디지털 회로 (IOI22_circuit) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "circuit.h"
#include <algorithm>
#include <iostream>
#include <vector>
using ull = unsigned long long;
const int MOD = 1000002022;
int N, M;
std::vector<int> P, A, prod;
std::vector<std::vector<int>> ch;
void init(int _N, int _M, std::vector<int> _P, std::vector<int> _A) {
N = _N; M = _M;
P = _P;
prod.resize(N + M);
std::fill(prod.begin() + N, prod.end(), 1);
for (int i = 1; i != N + M; ++i) ++prod[P[i]];
for (int i = N - 1; i; --i) prod[P[i]] = prod[P[i]] * (ull)prod[i] % P;
A.resize(N + M);
std::copy(_A.begin(), _A.end(), A.begin() + N);
ch.resize(N);
for (int i = 1; i != N + M; ++i) ch[P[i]].push_back(i);
}
int count_ways(int L, int R) {
for (int i = L; i <= R; ++i) A[i] = !A[i];
for (int i = N - 1; i >= 0; --i) {
int f0 = 1, f1 = 0;
for (int j : ch[i]) {
f1 = (f0 * (ull)A[j] + f1 * (ull)prod[j]) % MOD;
f0 = f0 * (ull)prod[j] % MOD;
}
A[i] = f1;
}
return A[0];
}