Submission #736331

#TimeUsernameProblemLanguageResultExecution timeMemory
736331jk410Digital Circuit (IOI22_circuit)C++17
18 / 100
12 ms8400 KiB
#include "circuit.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll MOD = 1000002022;

int n, m;
vector<int> p, a;
vector<int> child[2023];
ll dp[1023], dp2[2023][2023];
ll dpAll[1023];

ll getDPAll(int v) {
	ll& ret = dpAll[v] = (int)child[v].size();
	for (int i : child[v]) {
		if (i < n)
			ret = ret * getDPAll(i) % MOD;
	}
	return ret;
}

void init(int _n, int _m, vector<int> _p, vector<int> _a) {
	n = _n;
	m = _m;
	p = _p;
	a = _a;
	for (int i = 1; i < n + m; i++)
		child[p[i]].push_back(i);
	getDPAll(0);
}

void getDP(int v) {
	int sz = (int)child[v].size();
	for (int i : child[v]) {
		if (i < n)
			getDP(i);
	}
	dp2[0][0] = 1;
	for (int i = 1; i <= sz; i++) {
		int cur = child[v][i - 1];
		for (int j = 0; j <= i; j++) {
			dp2[i][j] = 0;
			if (cur < n)
				dp2[i][j] = dp2[i - 1][j] * ((dpAll[cur] - dp[cur] + MOD) % MOD) % MOD;
			else if (!a[cur - n])
				dp2[i][j] = dp2[i - 1][j];
		}
		for (int j = 1; j <= i; j++) {
			if (cur < n)
				dp2[i][j] = (dp2[i][j] + dp2[i - 1][j - 1] * dp[cur] % MOD) % MOD;
			else if (a[cur - n])
				dp2[i][j] = dp2[i - 1][j - 1];
		}
		dp2[i][i + 1] = 0;
	}
	dp[v] = 0;
	for (int i = 1; i <= sz; i++)
		dp[v] = (dp[v] + dp2[sz][i] * i % MOD) % MOD;
}

int count_ways(int l, int r) {
	for (int i = l - n; i <= r - n; i++)
		a[i] ^= 1;
	getDP(0);
	return dp[0];
}
#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...