#include <bits/stdc++.h>
#include "circuit.h"
using namespace std;
#define int long long
const int MXN = 2e5 + 5;
const int mod = 1e9 + 2022;
int n, m, res = 0;
vector<int> A, P;
vector<int> adj[MXN];
int dp[2][MXN], SZ[MXN];
int cnt = 0;
void calc(int a)
{
dp[0][a] = dp[1][a] = 0;
if (adj[a].empty())
{
dp[0][a] = A[a] ^ 1;
dp[1][a] = A[a];
return;
}
int sz = adj[a].size();
vector<int> dp1(sz + 1, 0);
dp1[0] = 1;
for (int &v : adj[a])
{
for (int i = sz; i >= 0; i--)
{
dp1[i] = (dp1[i] * dp[0][v]) % mod;
if (i) dp1[i] = (dp1[i] + ((dp1[i - 1] * dp[1][v]) % mod)) % mod;
}
}
for (int i = 0; i <= sz; i++)
{
dp[0][a] = (dp[0][a] + (((sz - i) * dp1[i]) % mod)) % mod;
dp[1][a] = (dp[1][a] + ((i * dp1[i]) % mod)) % mod;
}
}
void dfs(int a)
{
for (int &v : adj[a])
{
dfs(v);
}
calc(a);
}
int val[MXN];
void init(int32_t N, int32_t M, vector<int32_t> PP, vector<int32_t> AA)
{
n = N, m = M;
for (int32_t &i : PP) P.push_back(i);
for (int32_t &i : AA) A.push_back(i);
reverse(A.begin(), A.end());
A.resize(N + M, 0);
reverse(A.begin(), A.end());
for (int i = 1; i < N + M; i++) adj[P[i]].push_back(i);
dfs(0);
int org = dp[1][0];
for (int i = N; i < N + M; i++)
{
A[i] ^= 1;
dfs(0);
val[i] = abs(dp[1][0] - org);
A[i] ^= 1;
}
}
int32_t count_ways(int32_t L, int32_t R) {
for (int i = L; i <= R; i++) A[i] ^= 1;
int res = 0;
for (int i = n; i < n + m; i++)
{
res = (res + A[i] * val[i]) % mod;
}
return res;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
9048 KB |
Output is correct |
2 |
Correct |
1 ms |
9048 KB |
Output is correct |
3 |
Execution timed out |
3065 ms |
9048 KB |
Time limit exceeded |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
9048 KB |
Output is correct |
2 |
Incorrect |
5 ms |
9048 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '792435312' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
9048 KB |
Output is correct |
2 |
Correct |
1 ms |
9048 KB |
Output is correct |
3 |
Execution timed out |
3065 ms |
9048 KB |
Time limit exceeded |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Execution timed out |
3020 ms |
13908 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Execution timed out |
3020 ms |
13908 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
9048 KB |
Output is correct |
2 |
Incorrect |
5 ms |
9048 KB |
1st lines differ - on the 1st token, expected: '52130940', found: '792435312' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
9048 KB |
Output is correct |
2 |
Correct |
1 ms |
9048 KB |
Output is correct |
3 |
Execution timed out |
3065 ms |
9048 KB |
Time limit exceeded |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
9048 KB |
Output is correct |
2 |
Correct |
1 ms |
9048 KB |
Output is correct |
3 |
Execution timed out |
3065 ms |
9048 KB |
Time limit exceeded |
4 |
Halted |
0 ms |
0 KB |
- |