Submission #637190

#	Time	Username	Problem	Language	Result	Execution time	Memory
637190		blue	Digital Circuit (IOI22_circuit)	C++17	13 / 100	3026 ms	11980 KiB

circuit

#include "circuit.h"
#include <bits/stdc++.h>
using namespace std;

using vi = vector<int>;
using vvi = vector<vi>;
using ll = long long;
using vll = vector<ll>;
using vvll = vector<vll>;

const ll mod = 1'000'002'022;

ll ad(ll a, ll b)
{
	return (a+b)%mod;
}

ll mul(ll a, ll b)
{
	return (a*b)%mod;
}

const int mx = 200'000;

vll pow2(1 + mx);

vi children[1 + mx];
int N, M;

vi depth(1 + mx, 0);

vi A;

ll res = 0;

void dfs(int u)
{
	for(int v : children[u])
	{
		depth[v] = depth[u] + 1;
		dfs(v);
	}
}

void init(int N_, int M_, vi P_, vi A_)
{
	N = N_;
	M = M_;

	for(int i = 1; i < N+M; i++)
		children[P_[i]].push_back(i);

	dfs(0);

	A = vi(N, -1);
	for(int a : A_)
		A.push_back(a);

	pow2[0] = 1;
	for(int e = 1; e <= mx; e++)
		pow2[e] = mul(2, pow2[e-1]);

	// cerr << depth[1] << ' ' << depth[2] << '\n';

	for(int i = N; i < N+M; i++)
		res = ad(res, mul(A[i], pow2[N - depth[i]]));

	// cerr << "res = " << res << '\n';

	for(int f : A)
		cerr << f << '\n';
}

int count_ways(int L, int R)
{
	for(int i = L; i <= R; i++)
	{
		if(A[i] == 0)
		{
			// cerr << "activate " << i << "\n";
			A[i] = 1;
			res = ad(res, pow2[N - depth[i]]);
		}
		else
		{
			A[i] = 0;
			// cerr << "deactivate " << i << '\n';
			res = ad(res, mod - pow2[N - depth[i]]);
		}
	}

	return res;
}

• Subtask 1: 2 / 2
• Subtask 2: 7 / 7
• Subtask 3: 0 / 9
• Subtask 4: 4 / 4
• Subtask 5: 0 / 12
• Subtask 6: 0 / 27
• Subtask 7: 0 / 28
• Subtask 8: 0 / 11