Submission #1331445

#TimeUsernameProblemLanguageResultExecution timeMemory
1331445NonozeDigital Circuit (IOI22_circuit)C++17
46 / 100
3072 ms7824 KiB
#include "circuit.h"
/*
*	Author: Nonoze
*	Created: Tuesday 03/03/2026
*/
#include <bits/stdc++.h>
using namespace std;

#ifndef DEBUG
	#define dbg(...)
#endif

// #define cout cerr << "OUT: "
#define endl '\n'
#define endlfl '\n' << flush
#define quit(x) return (void)(cout << x << endl)

template<typename T> void read(T& x) { cin >> x; }
template<typename T1, typename T2> void read(pair<T1, T2>& p) { read(p.first), read(p.second); }
template<typename T> void read(vector<T>& v) { for (auto& x : v) read(x); }
template<typename T1, typename T2> void read(T1& x, T2& y) { read(x), read(y); }
template<typename T1, typename T2, typename T3> void read(T1& x, T2& y, T3& z) { read(x), read(y), read(z); }
template<typename T1, typename T2, typename T3, typename T4> void read(T1& x, T2& y, T3& z, T4& zz) { read(x), read(y), read(z), read(zz); }
template<typename T> void print(vector<T>& v) { for (auto& x : v) cout << x << ' '; cout << endl; }

#define sz(x) (int)(x.size())
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define make_unique(v) sort(all(v)), v.erase(unique(all(v)), (v).end())
#define pb push_back
#define mp(a, b) make_pair(a, b)
#define fi first
#define se second
#define cmin(a, b) a = min(a, b)
#define cmax(a, b) a = max(a, b)
#define YES cout << "YES" << endl
#define NO cout << "NO" << endl
#define QYES quit("YES")
#define QNO quit("NO")

const int MOD = 1e9+2022, LOG=20;

template<typename T, auto &md> 
struct modint {
	#define tpU template<typename U>
	using V = conditional_t<sizeof(T) <= 4, uint64_t, __uint128_t>;
	make_unsigned_t<T> x;
	modint() : x(0) {}
	tpU modint(U y) : x(y < md ? y : (y %= md) >=0 ? y : y += md) {}
	operator T() const { return x; }
	modint operator-() const { return modint() - *this; }
	tpU modint operator+(U y) const { return modint(*this) += y; }
	tpU modint operator-(U y) const { return modint(*this) -= y; }
	tpU modint operator*(U y) const { return modint(*this) *= y; }
	tpU modint operator/(U y) const { return modint(*this) /= y; }
	tpU modint operator++(U y) { if (++x == md) x = 0; return *this; }
	tpU modint operator--(U y) { if (!x--) x = md-1; return *this; }
	modint& operator+=(modint y) { if ((x += y.x) >= md) x -= md; return *this; }
	modint& operator-=(modint y) { if ((x += md-y.x) >= md) x -= md; return *this; }
	modint& operator*=(modint y) { x = x * (V)y.x % md; return *this; }
	modint& operator/=(modint y) { return *this *= y.inv(); }
	tpU modint pow(U y) const {
		if (y < 0) return inv(pow(-y));
		modint res(1), t(*this);
		for (; y; y = y>>1 , t *= t) if (y & 1) res *= t;
		return res;
	}
	modint inv() const { return pow(md-2); }

	// precompute factorials
	static void precompute_fact(T n) {
		if (n < fact_.size()) return;
		T i=sz(fact_);
		fact_.resize(n+1);
		ifact_.resize(n+1);
		for (; i<=n; i++) {
			fact_[i] = V(fact_[i-1])*i % md;
			ifact_[i]= V(ifact_[i-1])*inv(i) % md;
		}
	}

	static modint combi(T k, T n) {
		if (n < 0 || k < 0 || k > n) return 0;
		return fact(n) * ifact(k) * ifact(n-k);
	}

	static modint inv(T x) { return modint(x).inv(); }
	tpU static modint pow(U x, U y) { return modint(x).pow(y); }
	static modint fact(T x) { precompute_fact(x); return fact_[x]; }
	static modint ifact(T x) { precompute_fact(x); return ifact_[x]; }

	static inline vector<T> fact_={1, 1}, ifact_={1, 1};
};

using mint = modint<uint32_t, MOD>;

vector<int> adj[200005], a, p;
vector<mint> nb, poss;
int n, m, root=0;

mint dfs(int u) {
	if (adj[u].empty()) return poss[u]=1;
	mint res=sz(adj[u]);
	for (auto &v: adj[u]) res*=dfs(v);
	return poss[u]=res;
}

mint calcmult(int u) {
	if (u==root) return 1;
	int par=p[u];
	mint res=1;
	for (auto &v: adj[par]) if (v!=u && v!=p[par]) {
		res*=poss[v];
	}
	return res*calcmult(par);
}

void init(int N, int M, vector<int> P, vector<int> aa) {
	a=aa, p=P;
	n=N, m=M; 
	for (int i=1; i<n+m; i++) adj[p[i]].push_back(i);
	poss.resize(n+m); dfs(root);
	for (int i=0; i<m; i++) nb.push_back(calcmult(i+n));
}

int count_ways(int L, int R) {
	for (int i=L; i<=R; i++) a[i-n]^=1;
	mint ans=0;
	for (int i=0; i<m; i++) {
		if (a[i]) {
			ans+=nb[i];
		}
	}
	return ans;
}
#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...