제출 #627699

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
627699		ash7728	Digital Circuit (IOI22_circuit)	C++17	18 / 100	3086 ms	4940 KiB

circuit

#include "circuit.h"
#include <bits/stdc++.h>
#define N 100020
#define f first
#define s second
#define pb push_back
#define mp make_pair
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
using namespace std;
const int mod = 1000002022;
 
int yes[N], no[N], state[N],pai[N], n,m;
// int sum_cost[]
vector<int> grafo[N];
 
void solve(int x){
	yes[x] = no[x] = 0;
	if(x >= n){
		yes[x] = state[x];
		no[x] = (1-state[x]);
		return;
	}
	vector<int> poly;
	int sum_cost = 1, C = sz(grafo[x]);
	for(auto v: grafo[x]){
		solve(v);
		int old_yes = yes[x];

		yes[x] = (1LL*(old_yes)*no[v])%mod;
		yes[x] += (1LL*(old_yes + sum_cost)*yes[v])%mod;
		yes[x] %= mod;


		// cout<<old_yes<<" xxxxxxxxx "<<sum_cost<<"\n";
		sum_cost = (1LL*sum_cost*(yes[v] + no[v]))%mod; 
	}

	no[x] = (1LL*C*sum_cost%mod - yes[x])%mod;
	no[x] = (no[x] + mod)%mod;
 }
 
void init(int n_, int m_, vector<int> P, vector<int> A) {
	n = n_, m = m_;
	for(int i = 1; i < n+m; i++){
		grafo[P[i]].pb(i);
		pai[i] = P[i];
	}
	for(int i = 0;i<m;i++) state[i+n] = A[i];
}
 
int count_ways(int L, int R) {
	for(int i = L; i <= R; i++) state[i] ^= 1;
  	solve(0);
  	return yes[0];
 
}

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