Submission #791559

#	Time	Username	Problem	Language	Result	Execution time	Memory
791559		WongChun1234	Digital Circuit (IOI22_circuit)	C++17	100 / 100	925 ms	40048 KiB

circuit

#include "circuit.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=200050;
const ll MOD=1000002022;
int n,m,lz[N<<2];
ll dp[N],rdp[N],seg[N<<2],iseg[N<<2];
vector<int> p,a;
vector<ll> ch[N],pref[N],suff[N];
void build(int id,int l,int r){
	if (l==r){
		seg[id]=a[l-1]*rdp[l+n-1];
		iseg[id]=(!a[l-1])*rdp[l+n-1];
		//cout<<id<<" "<<l<<" "<<r<<" "<<seg[id]<<" "<<iseg[id]<<"\n";
		return;
	}
	int m=(l+r)/2;
	build(id*2,l,m);
	build(id*2+1,m+1,r);
	seg[id]=(seg[id*2]+seg[id*2+1])%MOD;
	iseg[id]=(iseg[id*2]+iseg[id*2+1])%MOD;
	//cout<<id<<" "<<l<<" "<<r<<" "<<seg[id]<<" "<<iseg[id]<<"\n";
}
void pushdown(int id){
	lz[id]^=1;
	swap(seg[id],iseg[id]);
}
void update(int id,int l,int r,int ql,int qr){
	if (l>=ql&&r<=qr){
		pushdown(id);
		return;
	}
	if (l>qr||r<ql) return;
	if (lz[id]){
		pushdown(id*2);
		pushdown(id*2+1);
		lz[id]=0;
	}
	int m=(l+r)/2;
	update(id*2,l,m,ql,qr);
	update(id*2+1,m+1,r,ql,qr);
	seg[id]=(seg[id*2]+seg[id*2+1])%MOD;
	iseg[id]=(iseg[id*2]+iseg[id*2+1])%MOD;
}
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++) ch[p[i]].push_back(i);
	for (int i=n+m-1;i>=n;i--){
		dp[i]=1;
	}
	for (int i=n-1;i>=0;i--){
		pref[i].push_back(1),suff[i].push_back(1);
		for (int j:ch[i]) pref[i].push_back(dp[j]),suff[i].push_back(dp[j]);
		pref[i].push_back(1),suff[i].push_back(1);
		for (int j=1;j<=ch[i].size();j++) pref[i][j]=pref[i][j-1]*pref[i][j]%MOD;
		for (int j=ch[i].size();j>=1;j--) suff[i][j]=suff[i][j+1]*suff[i][j]%MOD;
		dp[i]=suff[i][1]*ch[i].size()%MOD;
	}
	rdp[0]=1;
	for (int i=0;i<n;i++){
		for (int j=1;j<=ch[i].size();j++) rdp[ch[i][j-1]]=rdp[i]*(pref[i][j-1]*suff[i][j+1]%MOD)%MOD;
	}
	//for (int i=0;i<n+m;i++) cout<<dp[i]<<" "<<rdp[i]<<"\n";
	build(1,1,m);
}

int count_ways(int L, int R) {
	update(1,1,m,L-n+1,R-n+1);
	return seg[1];
}

Compilation message (stderr)

circuit.cpp: In function 'void init(int, int, std::vector<int>, std::vector<int>)':
circuit.cpp:56:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   56 |   for (int j=1;j<=ch[i].size();j++) pref[i][j]=pref[i][j-1]*pref[i][j]%MOD;
      |                ~^~~~~~~~~~~~~~
circuit.cpp:62:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |   for (int j=1;j<=ch[i].size();j++) rdp[ch[i][j-1]]=rdp[i]*(pref[i][j-1]*suff[i][j+1]%MOD)%MOD;
      |                ~^~~~~~~~~~~~~~

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