Submission #406453

#	Time	Username	Problem	Language	Result	Execution time	Memory
406453		amunduzbaev	Painting Walls (APIO20_paint)	C++14	51 / 100	415 ms	468260 KiB

paint

#include "paint.h"
#include "bits/stdc++.h"
#ifndef EVAL
#include "grader.cpp"
#endif
using namespace std;

#define int long long
#define sz(x) (int)x.size()
#define pb push_back
#define pii pair<int, int>
#define ff first
#define ss second

const int mod = 1e9+7;
const int N = 2e4+5;
const int MM = 4e3+5;
const int M = 1e5+5;
int gg[MM][N], n, m, k, is[N];
vector<int> pos[M];
int tree[4*N], p[4*N];

void push(int x, int lx, int rx){
	if(!p[x] || lx == rx) return;
	tree[x<<1] += p[x], tree[x<<1|1] += p[x];
	p[x<<1] += p[x], p[x<<1|1] += p[x];
	p[x] = 0;
}

void sett(int l, int r, int v, int lx = 0, int rx = n-1, int x = 1){
	push(x, lx, rx);
	if(lx > r || rx < l) return;
	if(lx >= l && rx <= r) { p[x] += v, tree[x] += v; return; }
	int m = (lx + rx)>>1;
	sett(l, r, v, lx, m, x<<1), sett(l, r, v, m+1, rx, x<<1|1);
	tree[x] = min(tree[x<<1], tree[x<<1|1]);
}

int find(int l, int r, int lx = 0, int rx = n-1, int x = 1){
	push(x, lx, rx);
	if(lx > r || rx < l) return mod;
	if(lx >= l && rx <= r) return tree[x];
	int m = (lx + rx)>>1;
	return min(find(l, r, lx, m, x<<1), find(l, r, m+1, rx, x<<1|1));
}

/*

8 3 5
3 3 1 3 4 4 2 2
3 0 1 2
2 2 3
2 3 4

*/

void check(){
	assert(n <= N && 2*m <= MM && k <= M);
}

int32_t minimumInstructions(int32_t N, int32_t M, int32_t K, vector<int32_t> c,
	vector<int32_t> a, vector<vector<int32_t>> b) {
	n = N, m = M, k = K;
	check();
	for(int i=0;i<n;i++) pos[c[i]].pb(i);
	for(int i=0;i<m;i++){
		for(int j=0;j<a[i];j++){
			for(auto x : pos[b[i][j]]) gg[i][x] = 1;
		}
	} for(int i=0;i<m;i++){
		for(int j=0;j<a[i];j++){
			for(auto x : pos[b[i][j]]) gg[i+m][x] = 1;
		}
	}
	for(int i=0;i<2*m;i++){
		for(int j=0;j<n;j++){
			if(i > 0 && j > 0 && gg[i][j]) gg[i][j] += gg[i-1][j-1];
			if(gg[i][j] >= m) is[j - m + 1] = 1;
		}
	} int res = 0;
	for(int i=0;i<n;i++) if(is[i]) sett(i, i+m-1, 1), res++;
	if(tree[1] == 0) return -1;
	for(int i=0;i<n;i++){
		if(!is[i]) continue;
		if(find(i, i+m-1) >= 2) sett(i, i+m-1, -1), res--;
	} 
	//for(int i=0;i<n;i++) cout<<is[i]<<" ";
	//cout<<"\n";
	//int res2 = 0;
	//memset(tree, 0, sizeof tree);
	//memset(p, 0, sizeof p);
	//for(int i=0;i<n;i++) if(is[i]) sett(i, i+m-1, 1), res2++;
	//for(int i=n-1;i>=0;i--){
		//if(!is[i]) continue;
		//if(find(i, i+m-1) >= 2) sett(i, i+m-1, -1), res2--;
	//} 
	//cout<<res<<" "<<res2<<"\n";
	
	return res;
}

• Subtask 1: 0 / 12
• Subtask 2: 15 / 15
• Subtask 3: 13 / 13
• Subtask 4: 23 / 23
• Subtask 5: 0 / 37