Submission #1058016

#	Time	Username	Problem	Language	Result	Execution time	Memory
1058016		tolbi	Digital Circuit (IOI22_circuit)	C++17	18 / 100	3079 ms	3160 KiB

circuit

#include "circuit.h"

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

constexpr int MOD = 1e9+2022;
vector<vector<int>> arr;
int N,M,P;
array<int,2> dp[200023];
void init(int N, int M, std::vector<int> P, std::vector<int> A) {
	::N=N,::M=M;
	arr.resize(N);
	for (int i = 1; i < N+M; i++){
		arr[P[i]].push_back(i);
	}
	for (int i = N; i < N+M; i++){
		dp[i][A[i-N]]=1;
		dp[i][A[i-N]^1]=0;
	}
}

int count_ways(int L, int R) {
	for (int i = L; i <= R; i++){
		swap(dp[i][0],dp[i][1]);
	}
	for (int i = N-1; i >= 0; i--){
		vector<int> curdp(1,0);
		curdp[0]=1;
		for (int j = 0; j < arr[i].size(); j++){
			vector<int> ndp(curdp.size()+1);
			for (int k = 0; k < ndp.size(); k++){
				if (k!=0){
					ndp[k]=(1ll*curdp[k-1]*dp[arr[i][j]][1])%MOD;
					if (ndp[k]>=MOD) ndp[k]-=MOD;
				}
				if (k!=curdp.size()){
					ndp[k]+=(1ll*curdp[k]*dp[arr[i][j]][0])%MOD;
					if (ndp[k]>=MOD) ndp[k]-=MOD;
				}
			}
			swap(ndp,curdp);
		}
		dp[i][0]=dp[i][1]=0;
		for (int j = 0; j < curdp.size(); j++){
			dp[i][0]+=(1ll*(arr[i].size()-j)*curdp[j])%MOD;
			dp[i][1]+=(1ll*j*curdp[j])%MOD;
			if (dp[i][0]>=MOD) dp[i][0]-=MOD;
			if (dp[i][1]>=MOD) dp[i][1]-=MOD;
		}
	}
	return dp[0][1];
}

Compilation message (stderr)

circuit.cpp: In function 'int count_ways(int, int)':
circuit.cpp:29:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   29 |   for (int j = 0; j < arr[i].size(); j++){
      |                   ~~^~~~~~~~~~~~~~~
circuit.cpp:31:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |    for (int k = 0; k < ndp.size(); k++){
      |                    ~~^~~~~~~~~~~~
circuit.cpp:36:10: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   36 |     if (k!=curdp.size()){
      |         ~^~~~~~~~~~~~~~
circuit.cpp:44:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   44 |   for (int j = 0; j < curdp.size(); j++){
      |                   ~~^~~~~~~~~~~~~~

• 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