oj.uz

Submission #1077372

# Submission time Handle Problem Language Result Execution time Memory
1077372 2024-08-27T06:20:16 Z UmairAhmadMirza Digital Circuit (IOI22_circuit) C++17
2 / 100
 1434 ms 22360 KB 
#include "circuit.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
ll const N=3e5+5;
ll const mod=1000002022;
vector<ll> child[N];
ll on[N],off[N],tot[N];
vector<ll> dp[N];
ll n,m;
vector<int> par,st;
ll dep[N];
void dfs(ll node){
	for(ll i:child[node]){
		dep[i]=dep[node]+1;
		dfs(i);
	}
}
ll vis[N];

void compute(ll node){
	ll c=child[node].size();
	tot[node]=c;
	for(ll i:child[node])
		tot[node]*=tot[i];
	for (ll i = 0; i <=c+2; ++i)
	{
		dp[i].clear();
		dp[i].resize(c+5,0);
	}
	dp[1][1]=on[child[node][0]];
	dp[1][0]=off[child[node][0]];
	for(ll i=2;i<=c;i++){
		for(ll j=0;j<=i;j++)
		{
			dp[i][j]=(dp[i-1][j]*off[child[node][i-1]]%mod)%mod;
			if(j==0)
				continue;
			dp[i][j]+=((dp[i-1][j-1]*on[child[node][i-1]])%mod);
			dp[i][j]%=mod;
		}
		// cout<<endl;
	}
	on[node]=0;
	for(ll i=1;i<=c;i++)
		on[node]=(on[node]+((dp[c][i]*i)%mod))%mod;
	off[node]=(((tot[node]-on[node])%mod)+mod)%mod;
}

void init(int nn, int mm, vector<int> P, vector<int> A){
	// cerr<<"GOT"<<endl;
	n=nn;
	m=mm;
	par=P;
	st=A;
	for(ll i=1;i<n+m;i++)
		child[par[i]].push_back(i);
	dfs(0);
	priority_queue<pair<ll,ll>> q;
	for(ll i=n;i<n+m;i++){
		tot[i]=1;
		if(st[i-n]==1){
			on[i]=1;
			off[i]=0;
		}
		else{
			on[i]=0;
			off[i]=1;
		}
		// cout<<on[i]<<' '<<off[i]<<' '<<tot[i]<<endl;
		q.push({dep[par[i]],par[i]});
	}
	while(q.size()){
		ll node=(q.top()).second;
		q.pop();
		if(node==-1 || vis[node]>=2)
			continue;
		vis[node]++;
		compute(node);
		q.push({dep[par[node]],par[node]});
	}
	// for (ll i = 0; i < n; ++i)
		// cout<<on[i]<<' '<<off[i]<<' '<<tot[i]<<endl;
}
int count_ways(int L, int R){
	// cout<<"L R := "<<L<<' '<<R<<endl;
	for (ll i = 0; i < n+m; ++i)
		vis[i]=0;
	priority_queue<pair<ll,ll>> q;
	for(ll i=L;i<=R;i++){
		st[i-n]^=1;
		tot[i]=1;
		if(st[i-n]==1){
			on[i]=1;
			off[i]=0;
		}
		else{
			on[i]=0;
			off[i]=1;
		}
		q.push({dep[par[i]],par[i]});
	}
	while(q.size()){
		ll node=(q.top()).second;
		q.pop();
		if(node==-1 || vis[node]>=2)
			continue;
		vis[node]++;
		// cout<<node<<endl;
		compute(node);
		q.push({dep[par[node]],par[node]});
	}
	// for (ll i = 0; i < n+m; ++i)
	// 	cout<<on[i]<<' '<<off[i]<<' '<<tot[i]<<endl;
	return on[0]%mod;
}

Subtask #1
2.0 / 2.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 14424 KB Output is correct
2 Correct 6 ms 14412 KB Output is correct
3 Correct 42 ms 22100 KB Output is correct
4 Correct 43 ms 22360 KB Output is correct
5 Correct 45 ms 22360 KB Output is correct
6 Correct 46 ms 22360 KB Output is correct
7 Correct 45 ms 22360 KB Output is correct
8 Correct 31 ms 22360 KB Output is correct

Subtask #2
0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 14424 KB Output is correct
2 Incorrect 8 ms 14520 KB 1st lines differ - on the 1st token, expected: '52130940', found: '0'
3 Halted 0 ms 0 KB -

Subtask #3
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 14424 KB Output is correct
2 Correct 6 ms 14412 KB Output is correct
3 Correct 42 ms 22100 KB Output is correct
4 Correct 43 ms 22360 KB Output is correct
5 Correct 45 ms 22360 KB Output is correct
6 Correct 46 ms 22360 KB Output is correct
7 Correct 45 ms 22360 KB Output is correct
8 Correct 31 ms 22360 KB Output is correct
9 Correct 6 ms 14424 KB Output is correct
10 Incorrect 8 ms 14520 KB 1st lines differ - on the 1st token, expected: '52130940', found: '0'
11 Halted 0 ms 0 KB -

Subtask #4
0 / 4.0

# Verdict Execution time Memory Grader output
1 Incorrect 1434 ms 19564 KB 1st lines differ - on the 1st token, expected: '431985922', found: '0'
2 Halted 0 ms 0 KB -

Subtask #5
0 / 12.0

# Verdict Execution time Memory Grader output
1 Incorrect 1434 ms 19564 KB 1st lines differ - on the 1st token, expected: '431985922', found: '0'
2 Halted 0 ms 0 KB -

Subtask #6
0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 14424 KB Output is correct
2 Incorrect 8 ms 14520 KB 1st lines differ - on the 1st token, expected: '52130940', found: '0'
3 Halted 0 ms 0 KB -

Subtask #7
0 / 28.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 14424 KB Output is correct
2 Correct 6 ms 14412 KB Output is correct
3 Correct 42 ms 22100 KB Output is correct
4 Correct 43 ms 22360 KB Output is correct
5 Correct 45 ms 22360 KB Output is correct
6 Correct 46 ms 22360 KB Output is correct
7 Correct 45 ms 22360 KB Output is correct
8 Correct 31 ms 22360 KB Output is correct
9 Correct 6 ms 14424 KB Output is correct
10 Incorrect 8 ms 14520 KB 1st lines differ - on the 1st token, expected: '52130940', found: '0'
11 Halted 0 ms 0 KB -

Subtask #8
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 14424 KB Output is correct
2 Correct 6 ms 14412 KB Output is correct
3 Correct 42 ms 22100 KB Output is correct
4 Correct 43 ms 22360 KB Output is correct
5 Correct 45 ms 22360 KB Output is correct
6 Correct 46 ms 22360 KB Output is correct
7 Correct 45 ms 22360 KB Output is correct
8 Correct 31 ms 22360 KB Output is correct
9 Correct 6 ms 14424 KB Output is correct
10 Incorrect 8 ms 14520 KB 1st lines differ - on the 1st token, expected: '52130940', found: '0'
11 Halted 0 ms 0 KB -