Submission #1367141

#TimeUsernameProblemLanguageResultExecution timeMemory
1367141Johan9월 (APIO24_september)C++20
100 / 100
723 ms34100 KiB
#include "september.h"
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 5;
const int LG = 20;
int tot = 0, tmr = 0, inf = 0, n;
map < int , int > mp;
int in[N], out[N], sub[N], cnt[N], up[N][LG];
int lzy[N * 4], lzy2[N * 4], st[N * 4], st2[N * 4], st3[N * 4];
vector < int > adj[N];
void dfs(int u, int p){
	in[u] = ++tmr;
	sub[u] = 1;
	up[u][0] = p;
	for(int i = 1; i < LG; i++)
		up[u][i] = up[up[u][i - 1]][i - 1];
	for(auto v : adj[u]){
		dfs(v, u);
		sub[u] += sub[v];
	}
	out[u] = tmr;
}
bool is_anc(int u, int v){
	return (in[u] <= in[v] && out[v] <= out[u]);
}
void upd(int v, int l, int r, int pos){
	if(l == r){
		st[v] = 1;
		return;
	}
	int mid = (l + r) >> 1;
	if(mid >= pos)upd(v * 2, l, mid, pos);
	else upd(v * 2 + 1, mid + 1, r, pos);
	st[v] = st[v * 2] + st[v * 2 + 1];
}
int ask(int v, int l, int r, int ql, int qr){
	if(l > qr || r < ql)return 0;
	if(l >= ql && r <= qr)return st[v];
	int mid = (l + r) >> 1;
	return ask(v * 2, l, mid, ql, qr) + ask(v * 2 + 1, mid + 1, r, ql, qr);
}
void relax(int v, int l, int r){
	if(lzy[v] == -1)return;
	if(l != r){
		lzy[v * 2] = lzy[v];
		lzy[v * 2 + 1] = lzy[v];
	}
	st2[v] = (r - l + 1) * lzy[v];
	lzy[v] = -1;
}
void upd2(int v, int l, int r, int ul, int ur, int x){
	relax(v, l, r);
	if(l > ur || r < ul)return;
	if(l >= ul && r <= ur){
		lzy[v] = x;
		relax(v, l, r);
		return;
	}
	int mid = (l + r) >> 1;
	upd2(v * 2, l, mid, ul, ur, x);
	upd2(v * 2 + 1, mid + 1, r, ul, ur, x);
	st2[v] = st2[v * 2] + st2[v * 2 + 1];
}
int ask2(int v, int l, int r, int ql, int qr){
	relax(v, l, r);
	if(l > qr || r < ql)return 0;
	if(l >= ql && r <= qr)return st2[v];
	int mid = (l + r) >> 1;
	return ask2(v * 2, l, mid, ql, qr) + ask2(v * 2 + 1, mid + 1, r, ql, qr);
}
void relax2(int v, int l, int r){
	if(lzy2[v] == -1)return;
	if(l != r){
		lzy2[v * 2] = lzy2[v];
		lzy2[v * 2 + 1] = lzy2[v];
	}
	st3[v] = (r - l + 1) * lzy2[v];
	lzy2[v] = -1;
}
void upd3(int v, int l, int r, int ul, int ur){
	relax2(v, l, r);
	if(l > ur || r < ul)return;
	if(l >= ul && r <= ur){
		lzy2[v] = 1;
		relax2(v, l, r);
		return;
	}
	int mid = (l + r) >> 1;
	upd3(v * 2, l, mid, ul, ur);
	upd3(v * 2 + 1, mid + 1, r, ul, ur);
	st3[v] = st3[v * 2] + st3[v * 2 + 1];
}
int ask3(int v, int l, int r, int ql, int qr){
	relax2(v, l, r);
	if(l > qr || r < ql)return 0;
	if(l >= ql && r <= qr)return st3[v];
	int mid = (l + r) >> 1;
	return ask3(v * 2, l, mid, ql, qr) + ask3(v * 2 + 1, mid + 1, r, ql, qr);
}
bool ok(int node){
	int anc = -1;
	node = up[node][0];
	if(node && ask3(1, 1, n, in[node], in[node]) >= 1){
		anc = node;
		for(int i = LG - 1; i >= 0; i--){
			int nxt = up[node][i];
			int cur = ask3(1, 1, n, in[nxt], in[nxt]);
			if(nxt && cur >= 1){
				node = up[node][i];
				anc = node;
			}
		}
	}
	if(anc == -1)
		return true;
	int tot = ask(1, 1, n, in[anc], out[anc]);
	if(sub[anc] == tot){
		inf -= ask2(1, 1, n, in[anc], out[anc]);
		upd2(1, 1, n, in[anc], out[anc], 0);
	}
	return (sub[anc] == tot);
}
int solve(int n_, int m, vector < int > F, vector < vector < int > > S){
	n = n_, tmr = 0, inf = 0;
	for(int i = 0; i <= n * 4; i++){
		lzy[i] = lzy2[i] = -1;
		st[i] = st2[i] = st3[i] = 0;
	}
	for(int i = 0; i <= n; i++){
		adj[i].clear();
		in[i] = out[i] = sub[i] = cnt[i] = 0;
		for(int j = 0; j < LG; j++)
			up[i][j] = 0;
	}
	for(int i = 0; i < n; i++){
		F[i]++;
		adj[F[i]].push_back(i + 1);
	}
	for(int i = 0; i < m; i++){
		for(int j = 0; j < S[i].size(); j++){
			S[i][j]++;
		}
	}
	dfs(1, 1);
	int k = 0, cntt = 0;
	for(int i = 0; i < n - 1; i++){
		for(int j = 0; j < m; j++){
			int node = S[j][i];
			cnt[node]++;
			if(cnt[node] == m){
				cntt++;
				upd3(1, 1, n, in[node], out[node]);
				upd(1, 1, n, in[node]);
				bool is1 = (ask(1, 1, n, in[node], out[node]) == sub[node]);
				bool is2 = ok(node);
				if(!is1 || !is2){
					upd2(1, 1, n, in[node], in[node], 1);
					inf++;	
				}
			}
		}
		k += (cntt == i + 1 && inf == 0);
	}
	return k;
}
/*
6 1
0 0 1 3 1
5 3 1 2 4
	*/
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...