Submission #1312257

#	Time	Username	Problem	Language	Result	Execution time	Memory
1312257		azamurai	September (APIO24_september)	C++20	30 / 100	1094 ms	6500 KiB

september

#include "september.h"
#include <bits/stdc++.h>
#include <vector>

using namespace std;

int get(int v, vector<int>& parent) {
	if (parent[v] == v) return v;
	return parent[v] = get(parent[v], parent);
}

void union_set(int u, int v, vector <int>& parent) {
	int pu = get(u, parent);
	int pv = get(v, parent);
	if (pu != pv) {
		parent[pu] = pv;
	}
}

int solve(int N, int M, std::vector<int> F, std::vector<std::vector<int>> S) {
	vector <int> parent(N + 1, 0);
	for (int i = 0; i < N; i++) {
		parent[i] = i;
	}
	for (int T = 0; T < M; T++) {
		vector <int> p(N + 1, 0);
		vector <int> pos(N, 0);
		for (int i = 0; i < N - 1; i++) {
			pos[S[T][i]] = i;
		}
		vector <pair<int,int>> seg;
		for (int i = 1; i < N; i++) {
			if (F[i] == 0) continue;
			if (pos[F[i]] < pos[i]) {
				seg.push_back(make_pair(pos[F[i]], pos[i]));
			}
		}
		sort(seg.begin(), seg.end());
		for (int i = 0; i < N; i++) {
			p[i] = i;
		}
		int idx = -1;
		for (auto to : seg) {
			int l = to.first, r = to.second;
			if (idx >= l) {
				for (int i = idx + 1; i <= r; i++) p[i] = p[i - 1];
				idx = r;
			}
			else {
				for (int i = l + 1; i <= r; i++) p[i] = p[i - 1];
				idx = r;
			}
		}
		set <int> st;
		for (int i = 0; i < N - 1; i++) {
			st.insert(p[i]);
		}
		for (auto to : st) {
			vector <int> save;
			for (int i = 0; i < N - 1; i++) {
				if (p[i] == to) save.push_back(S[T][i]);
			}
			for (int i = 1; i < (int)save.size(); i++) {
				union_set(save[i - 1], save[i], parent);
			}
		}
	}
	set <int> ans;
	for (int i = 1; i < N; i++) {
		ans.insert(get(i, parent));
	}
	return (int)ans.size();
}

• Subtask 1: 11 / 11
• Subtask 2: 0 / 14
• Subtask 3: 5 / 5
• Subtask 4: 9 / 9
• Subtask 5: 5 / 5
• Subtask 6: 0 / 11
• Subtask 7: 0 / 9
• Subtask 8: 0 / 11
• Subtask 9: 0 / 9
• Subtask 10: 0 / 16